Unverified Commit eac8af6a authored by vcherepanov-nv's avatar vcherepanov-nv Committed by GitHub
Browse files

Remove test skip logic for GEMM-AR tests (#2516)



* Use GEMM-AR fallback on newer cuBLASMp
Signed-off-by: default avatarVladimir Cherepanov <vcherepanov@nvidia.com>

* Remove test skip logic completely
Signed-off-by: default avatarVladimir Cherepanov <vcherepanov@nvidia.com>

---------
Signed-off-by: default avatarVladimir Cherepanov <vcherepanov@nvidia.com>
parent 2886cbce
...@@ -63,12 +63,6 @@ int main(int argc, char* argv[]) { ...@@ -63,12 +63,6 @@ int main(int argc, char* argv[]) {
return ret; return ret;
} }
bool IsMulticastSupported(int device_id) {
int supported = 0;
CHECK_CU(cuDeviceGetAttribute(&supported, CU_DEVICE_ATTRIBUTE_MULTICAST_SUPPORTED, device_id));
return supported;
}
int GetDeviceComputeCapability(int device_id) { int GetDeviceComputeCapability(int device_id) {
int major{}; int major{};
int minor{}; int minor{};
...@@ -369,11 +363,6 @@ struct GemmAr : public CommGemmFixure { ...@@ -369,11 +363,6 @@ struct GemmAr : public CommGemmFixure {
nvte_gemm_all_reduce(ctx_, m, n, k, a, b, d, bias, pre_act_out, transa, transb, grad, nvte_gemm_all_reduce(ctx_, m, n, k, a, b, d, bias, pre_act_out, transa, transb, grad,
accumulate, comm_sm_count, stream, kNVTECommGemmAlgoDefault); accumulate, comm_sm_count, stream, kNVTECommGemmAlgoDefault);
} }
void SetUp() override {
if (!IsMulticastSupported(rank_))
GTEST_SKIP() << "Multicast is not supported on device " << rank_;
}
}; };
TEST_P(AgGemm, Gemm) { TEST_P(AgGemm, Gemm) {
......
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