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

moved the architecture check to IsSupported function

Browse Source
This commit is contained in:
illsilin
2025-03-24 20:18:52 -07:00
parent 6e9e067f9b
commit 056d2a08b3
3 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle_v3_b_preshuffle.hpp
View File

@@ -379,7 +379,7 @@ struct DeviceGemm_Xdl_CShuffleV3_BPreshuffle : public DeviceGemmV2BPreshuffle<AL
static bool IsSupportedArgument(const Argument& arg)
{
if(!ck::is_xdl_supported())
if(!is_bf16_atomic_supported())
{
return false;
}