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

update kernel

Browse Source
This commit is contained in:
kyle-256
2025-12-18 07:49:20 +00:00
committed by kyle-256
parent 3053fb50ef
commit c08987af05
1 changed files with 7 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
example/ck_tile/17_grouped_gemm/grouped_gemm.cpp
View File

@@ -218,17 +218,9 @@ float grouped_gemm_tileloop(const ck_tile::stream_config& s,
num_groups));
};
if(!splitk)
{
return ave_time = Run(ck_tile::integral_constant<ck_tile::memory_operation_enum,
ck_tile::memory_operation_enum::set>{});
}
else
{
return ave_time =
Run(ck_tile::integral_constant<ck_tile::memory_operation_enum,
ck_tile::memory_operation_enum::atomic_add>{});
}
(void)splitk; // splitk not supported
return ave_time = Run(ck_tile::integral_constant<ck_tile::memory_operation_enum,
ck_tile::memory_operation_enum::set>{});
}
#include "run_grouped_gemm_example.inc"
@@ -300,12 +292,7 @@ int run_grouped_gemm_example(int argc, char* argv[])
const std::string c_layout = arg_parser.get_str("c_layout");
const std::string data_type = arg_parser.get_str("prec");
if(data_type == "fp16")
{
return run_gemm_example_prec_type<GemmConfig<ck_tile::half_t>, ck_tile::half_t>(
a_layout, b_layout, c_layout, argc, argv);
}
else if(data_type == "bf16")
if(data_type == "bf16")
{
return run_gemm_example_prec_type<GemmConfig<ck_tile::bf16_t>, ck_tile::bf16_t>(
a_layout, b_layout, c_layout, argc, argv);
@@ -317,15 +304,13 @@ int run_grouped_gemm_example(int argc, char* argv[])
}
else
{
throw std::runtime_error("Unsupported data type configuration.");
throw std::runtime_error("Only bf16/fp8 data types are supported.");
}
}
int main(int argc, char* argv[])
{
#if CK_TILE_USE_WMMA
return !run_grouped_gemm_example<GemmConfigComputeV4_Wmma>(argc, argv);
#else
return run_grouped_gemm_example<GemmConfigComputeV3_2>(argc, argv);
#endif
}