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

paged window run ok

Browse Source
This commit is contained in:
coderfeli
2025-04-05 13:45:46 +00:00
parent 45a0463f1f
commit fe2ea699e5
10 changed files with 805 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
example/ck_tile/XX_moe_gemm/run_moe_gemm_example.inc
View File

@@ -254,22 +254,22 @@ int run_moe_gemm_example_with_layouts(int argc,
K, 1 /*kbatch*/, max_accumulated_value);
c_m_n_ref_buf->FromDevice(c_m_n_host_ref.data());
for(int im = 0; im < M; im++)
{
for(int in = 0; in < N; in++)
{
// if (static_cast<float>(static_cast<CDataType*>(p_c)[im * N + in]) != 0)
printf("c[%d][%d]: %f ",
im,
in,
static_cast<float>(static_cast<CDataType*>(p_c)[im * N + in]));
printf("ref[%d][%d]: %f \n",
im,
in,
static_cast<float>(
static_cast<CDataType*>(c_m_n_host_ref.data())[im * N + in]));
}
}
// for(int im = 0; im < M; im++)
// {
// for(int in = 0; in < N; in++)
// {
// // if (static_cast<float>(static_cast<CDataType*>(p_c)[im * N + in]) != 0)
// printf("c[%d][%d]: %f ",
// im,
// in,
// static_cast<float>(static_cast<CDataType*>(p_c)[im * N + in]));
// printf("ref[%d][%d]: %f \n",
// im,
// in,
// static_cast<float>(
// static_cast<CDataType*>(c_m_n_host_ref.data())[im * N + in]));
// }
// }
pass = ck_tile::check_err(c_m_n_tensor,
c_m_n_host_ref,