diff --git a/benchmarks/dense_matmul/gendata.py b/benchmarks/dense_matmul/gendata.py index 5cffc4ed..61afc6d6 100755 --- 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 index f55c6106..94f54dc7 100755 --- 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__":