Disable splitk for 2stage xdl on rdna (bug to be fixed)

2026-07-01 04:07:56 +00:00 · 2025-10-06 12:32:01 +00:00
parent 9eece6c0c4
commit 4c09ae57bc
1 changed files with 5 additions and 0 deletions
--- a/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_weight_two_stage_xdl_cshuffle.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_bwd_weight_two_stage_xdl_cshuffle.hpp
@@ -1745,6 +1745,11 @@ struct DeviceGroupedConvBwdWeightTwoStage_Xdl_CShuffle
            {
                return false;
            }
+            // TODO: this is needed because there is a bug
+            if(arg.k_batch_ > 1)
+            {
+                return false;
+            }
        }

        // Check this here, it allows to use other instances from factory even