Files
tinygrad/extra/optimization/extract_dataset.py
chenyu 207bca6cea set PAGE_SIZE=1 and generate new dataset (#7559)
13080 rows in total. both generating and loading this are pretty broken now. filters are wrong for example
2024-11-05 11:25:01 -05:00

14 lines
373 B
Python
Executable File

#!/usr/bin/env python3
# extract asts from process replay artifacts
import os
from test.external.process_replay.process_replay import _pmap
LOGOPS = os.getenv("LOGOPS", "/tmp/sops")
def extract_ast(*args) -> bool:
open(LOGOPS, "a").write(str(args[0]).replace("\n", "").replace(" ", "")+"\n")
return args[-1]
if __name__ == "__main__":
_pmap("kernel", extract_ast)