[tuner] Temporarily disable candidate pickling (#583)

`ir.*` types don't support pickling. Disabling this for now until we get a proper fix.
nod-ai · Nov 21, 2024 · 06599e9 · 06599e9
1 parent 5348a11
commit 06599e9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tuner/tuner/candidate_gen.py b/tuner/tuner/candidate_gen.py
@@ -554,8 +554,9 @@ def tune(
             with open(path.join(output, f"{i+1}_config.mlir"), "w") as f:
                 f.write(tf_mlir.embeddable)
 
-        with open(path.join(output, "configs.pkl"), "wb") as file:
-            pickle.dump(configs, file)
+        # TODO: Fix pickling for ir types.
+        # with open(path.join(output, "configs.pkl"), "wb") as file:
+        #    pickle.dump(configs, file)
 
         tune_logger.info(f"Generated {len(configs)} candidates")
         tune_logger.info(f"Configurations .pkl is stored in {output}/configs.pkl")