IQ.Pilot Release Commit @ 5bc9cd3

tinygrad_repo/extra/optimization/extract_dataset.py

@@ -0,0 +1,13 @@
+#!/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) -> None:
+  open(LOGOPS, "a").write(str(args[1]).replace("\n", "").replace(" ", "")+"\n")
+  return None
+
+if __name__ == "__main__":
+  _pmap({"do_to_program":extract_ast})