Commit 951ecad1 authored by Yanxing-Shi's avatar Yanxing-Shi
Browse files

Merge commit 'refs/pull/597/head' of...

Merge commit 'refs/pull/597/head' of https://github.com/ROCmSoftwarePlatform/composable_kernel into HEAD
parents 6eca23d7 e5fb9fa6
......@@ -105,7 +105,7 @@ struct DeviceGemmMultipleD_Wmma_CShuffle : public DeviceGemmMultipleD<ALayout,
#endif
}();
if constexpr(GemmSpec == GemmSpecialization::MNPadding)
if constexpr(GemmSpec == GemmSpecialization::MNKPadding)
{
const auto PadM = (MPerBlock - M % MPerBlock) % MPerBlock;
......@@ -144,7 +144,7 @@ struct DeviceGemmMultipleD_Wmma_CShuffle : public DeviceGemmMultipleD<ALayout,
}
}();
if constexpr(GemmSpec == GemmSpecialization::MNPadding)
if constexpr(GemmSpec == GemmSpecialization::MNKPadding)
{
const auto PadN = (NPerBlock - N % NPerBlock) % NPerBlock;
......@@ -180,7 +180,7 @@ struct DeviceGemmMultipleD_Wmma_CShuffle : public DeviceGemmMultipleD<ALayout,
}
}();
if constexpr(GemmSpec == GemmSpecialization::MNPadding)
if constexpr(GemmSpec == GemmSpecialization::MNKPadding)
{
const auto PadM = (MPerBlock - M % MPerBlock) % MPerBlock;
const auto PadN = (NPerBlock - N % NPerBlock) % NPerBlock;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment