diff --git a/tests/unittest/_torch/modules/test_fused_moe.py b/tests/unittest/_torch/modules/test_fused_moe.py index b33bb41886d0..49a7e30e9ba9 100644 --- a/tests/unittest/_torch/modules/test_fused_moe.py +++ b/tests/unittest/_torch/modules/test_fused_moe.py @@ -639,7 +639,6 @@ def set_tensor_value_4(x, num_row, num_cols): x.copy_(repeated) -@pytest.mark.skip(reason="https://nvbugs/5565565") @skip_pre_blackwell @pytest.mark.skipif(torch.cuda.device_count() < 4, reason="needs 4 GPUs to run this test")