From b3caa676949c856093468aa4b96f8c523cc2f58f Mon Sep 17 00:00:00 2001
From: Yanxing-Shi <yanxishi@amd.com>
Date: Mon, 19 May 2025 11:44:26 +0000
Subject: [PATCH] fix csv bug

---
 tile_engine/ops/gemm/gemm_host_api.cpp        |  4 ++--
 tile_engine/ops/gemm/gemm_instance_builder.py | 20 +++++++++----------
 tile_engine/ops/gemm/gemm_profiler.hpp        |  6 +++---
 3 files changed, 15 insertions(+), 15 deletions(-)

diff --git a/tile_engine/ops/gemm/gemm_host_api.cpp b/tile_engine/ops/gemm/gemm_host_api.cpp
index 3afa3195da..226dd511d8 100755
--- a/tile_engine/ops/gemm/gemm_host_api.cpp
+++ b/tile_engine/ops/gemm/gemm_host_api.cpp
@@ -7,7 +7,7 @@
 #include "gemm_host_api.hpp"
 #include "benchmark_gemm.hpp"
 
-auto run_single_trait(const ck_tile::ArgParser& arg_parser)
+auto get_kernel_func(const ck_tile::ArgParser& arg_parser)
 {
     KernelTraits trait;
     trait.pipeline  = arg_parser.get_str("pipeline");
@@ -29,7 +29,7 @@ int main(int argc, char* argv[])
         auto [result, parser] = create_args(argc, argv);
         if(!result)
             return EXIT_FAILURE;
-        benchmark_gemm(parser, run_single_trait(parser));
+        benchmark_gemm(parser, get_kernel_func(parser));
         return 0;
     }
     catch(const std::exception& e)
diff --git a/tile_engine/ops/gemm/gemm_instance_builder.py b/tile_engine/ops/gemm/gemm_instance_builder.py
index ba68727609..28c197d6ce 100755
--- a/tile_engine/ops/gemm/gemm_instance_builder.py
+++ b/tile_engine/ops/gemm/gemm_instance_builder.py
@@ -283,16 +283,16 @@ struct GemmKernel {{
     }}
 
     static std::string get_name() {{
-        return std::string("GemmKernel<Bllktile: ") + std::to_string(TileM) + "x" + std::to_string(TileN) + "x" + std::to_string(TileK) + ", " +
-                "WaveMap: " + std::to_string(WarpM) + "x" + std::to_string(WarpN) + "x" + std::to_string(WarpK) + ", " +
-                "WarpTile: " + std::to_string(WarpTileM) + "x" + std::to_string(WarpTileN) + "x" + std::to_string(WarpTileK) + ", " +
-                "PadidngM: " + "{pad_m}" + ", " +
-                "PaddingN: " + "{pad_n}" + ", " +
-                "PaddingK: " + "{pad_k}" + ", " +
-                "Pipeline: " + "{pipeline}" + ", " +
-                "Epilogue: " + "{epilogue}" + ", " +
-                "Scheduler: " + "{scheduler}";
-                }}
+        return std::string("gemm_") + std::to_string(TileM) + "x" + std::to_string(TileN) + "x" + std::to_string(TileK) +
+                "_" + std::to_string(WarpM) + "x" + std::to_string(WarpN) + "x" + std::to_string(WarpK) + "_" +
+                std::to_string(WarpTileM) + "x" + std::to_string(WarpTileN) + "x" + std::to_string(WarpTileK) + "_" +
+                "{pad_m}" + "_" +
+                "{pad_n}" + "_" +
+                "{pad_k}" + "_" +
+                "{pipeline}" + "_" +
+                "{epilogue}" + "_" +
+                "{scheduler}";
+    }}
 }};
 """
 
diff --git a/tile_engine/ops/gemm/gemm_profiler.hpp b/tile_engine/ops/gemm/gemm_profiler.hpp
index 5a9fa74316..14d65e90e4 100644
--- a/tile_engine/ops/gemm/gemm_profiler.hpp
+++ b/tile_engine/ops/gemm/gemm_profiler.hpp
@@ -84,7 +84,7 @@ class GemmProfiler
 
         if(!setting_.csv_filename_.empty())
         {
-            std::ofstream file(setting_.csv_filename_, std::ios::app);
+            std::ofstream file(setting_.csv_filename_ + ".csv", std::ios::app);
 
             if(!file.is_open())
             {
@@ -94,7 +94,7 @@ class GemmProfiler
             {
                 if(file.tellp() == 0)
                 {
-                    file << "rocm_version, device_name,"
+                    file << "rocm_version,device_name,"
                          << "split_k,m,n,k,stride_a,stride_b,stride_c,"
                          << "dtype_a,dtype_b,dtype_acc,dtype_c,"
                          << "layout_a,layout_b,layout_c,"
@@ -129,7 +129,7 @@ class GemmProfiler
         return kernel_instance;
     }
 
-    GemmProfiler(const GemmProfiler&)            = delete;
+    GemmProfiler(const GemmProfiler&) = delete;
     GemmProfiler& operator=(const GemmProfiler&) = delete;
 
     private: