Fix gendata.py

KULeuven-MICAS · Dec 20, 2024 · 41d1d55 · 41d1d55
1 parent e79b968
commit 41d1d55
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/benchmarks/dense_matmul/gendata.py b/benchmarks/dense_matmul/gendata.py
@@ -46,8 +46,8 @@ def create_test_data(m, n, k, add_c: bool = False):
         "D": D,
     }
 
-    create_header("data.h", sizes, variables)
-    create_data("data.c", variables)
+    create_header("data", sizes, variables)
+    create_data("data", variables)
 
 
 if __name__ == "__main__":

diff --git a/benchmarks/tiled_matmul/gendata.py b/benchmarks/tiled_matmul/gendata.py
@@ -67,8 +67,8 @@ def create_test_data(n, m, k, ones=False, random_shape=False):
         "C": C,
     }
 
-    create_header("data.h", sizes, variables)
-    create_data("data.c", variables)
+    create_header("data", sizes, variables)
+    create_data("data", variables)
 
 
 if __name__ == "__main__":