Skip to content

Commit eac8af6

Browse files
Remove test skip logic for GEMM-AR tests (#2516)
* Use GEMM-AR fallback on newer cuBLASMp Signed-off-by: Vladimir Cherepanov <vcherepanov@nvidia.com> * Remove test skip logic completely Signed-off-by: Vladimir Cherepanov <vcherepanov@nvidia.com> --------- Signed-off-by: Vladimir Cherepanov <vcherepanov@nvidia.com>
1 parent 2886cbc commit eac8af6

File tree

1 file changed

+0
-11
lines changed

1 file changed

+0
-11
lines changed

tests/cpp_distributed/test_comm_gemm.cu

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -63,12 +63,6 @@ int main(int argc, char* argv[]) {
6363
return ret;
6464
}
6565

66-
bool IsMulticastSupported(int device_id) {
67-
int supported = 0;
68-
CHECK_CU(cuDeviceGetAttribute(&supported, CU_DEVICE_ATTRIBUTE_MULTICAST_SUPPORTED, device_id));
69-
return supported;
70-
}
71-
7266
int GetDeviceComputeCapability(int device_id) {
7367
int major{};
7468
int minor{};
@@ -369,11 +363,6 @@ struct GemmAr : public CommGemmFixure {
369363
nvte_gemm_all_reduce(ctx_, m, n, k, a, b, d, bias, pre_act_out, transa, transb, grad,
370364
accumulate, comm_sm_count, stream, kNVTECommGemmAlgoDefault);
371365
}
372-
373-
void SetUp() override {
374-
if (!IsMulticastSupported(rank_))
375-
GTEST_SKIP() << "Multicast is not supported on device " << rank_;
376-
}
377366
};
378367

379368
TEST_P(AgGemm, Gemm) {

0 commit comments

Comments
 (0)