BENCH: auto-rm pk_cpp folder..

kokkos · Mar 19, 2023 · ab986bf · ab986bf
1 parent 520f7ee
commit ab986bf
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/benchmarks/dgemm_compare.py b/benchmarks/dgemm_compare.py
@@ -3,6 +3,7 @@
 """
 
 import os
+import shutil
 import time
 
 import pykokkos as pk
@@ -39,6 +40,10 @@
     if num_threads is None:
         raise ValueError("must set OMP_NUM_THREADS for benchmarks!")
 
+    cwd = os.getcwd()
+    shutil.rmtree(os.path.join(cwd, "pk_cpp"),
+                  ignore_errors=True)
+
     df = pd.DataFrame(np.full(shape=(num_global_repeats, 2), fill_value=np.nan),
                       columns=["scenario", "time (s)"])
     df["scenario"] = df["scenario"].astype(str)