ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-01 12:17:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix csv bug

Browse Source
This commit is contained in:
Yanxing-Shi
2025-05-19 11:44:26 +00:00
parent 9897410acf
commit b3caa67694
3 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tile_engine/ops/gemm/gemm_host_api.cpp
View File

@@ -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)

20
tile_engine/ops/gemm/gemm_instance_builder.py
View File

@@ -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}";
}}
}};
"""

6
tile_engine/ops/gemm/gemm_profiler.hpp
View File

@@ -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: