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

update CE elementOP

Browse Source
This commit is contained in:
mtgu0705
2025-05-13 02:19:13 -05:00
parent 5ba86c210b
commit 5b26ad3bbf
1 changed files with 0 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
example/67_gemm_microscaling/moe_gemm2_xdl_mx_fp4.cpp
View File

@@ -68,11 +68,7 @@ struct MulABScaleExpertWeight
(void)d1;
(void)d2;
#if CK_USE_PK4_LAYOUT_SHUFFLE
e = ck::type_convert<EDataType>(c);
#else
e = ck::type_convert<EDataType>(c);
#endif
}
// for reference cpu
template <>
@@ -80,11 +76,7 @@ struct MulABScaleExpertWeight
float& e, const float& c, const float& d0, const float& d1, const float& d2) const
{
// for reference cpu
#if CK_USE_PK4_LAYOUT_SHUFFLE
e = ck::type_convert<EDataType>(c * d0 * d1 * d2);
#else
e = ck::type_convert<EDataType>(c * d0 * d1 * d2);
#endif
}
};