progress

app.py

@@ -23,7 +23,7 @@ def generate_kernel(text_input, n_iters, progress=gr.Progress()):
     best_code = ""
     for info in run_autotune_pipeline(src_path, n_iters):
         # 1) update progress bar (if iteration known)
-        if info["iteration"] is not None:
+        if info["event"] ==  "iteration_end" and info["iteration"] is not None:
             # print(f"Iteration {info['iteration']} / {n_iters}: {info['message']}")
             progress((info["iteration"], n_iters), desc=info["message"])
 

tool/testv3.so

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9911ef353f67e6f1bd1ea5c7fea5e9bea863983dfbe8945069079343a25a4be3
+oid sha256:532ed0f05a260ea9260928e1ee4cfaee08d364dced2b18aec97a8e2956a59c1e
 size 487120