check gridwise level validity in device impl for 1 stage D0

This commit is contained in:
Enrico Degregori
2025-07-29 10:44:22 +00:00
parent 6514b15fb5
commit 2ba2c5d1b4

View File

@@ -1120,7 +1120,7 @@ struct DeviceGroupedConvBwdWeight_Wmma_CShuffleV3
}
}
return true;
return GridwiseGemm::CheckValidity(gemm_arg);
}
bool IsSupportedArgument(const BaseArgument* p_arg) override