Check NCCL existence in test decorators
@@ -15,14 +15,16 @@ class TestNCCL(unittest.TestCase):
comm.destroy()
@attr.gpu
- @unittest.skipUnless(cuda.nccl.get_version() >= 2400, "Using old NCCL")
+ @unittest.skipUnless(cuda.nccl_enabled and
+ cuda.nccl.get_version() >= 2400, "Using old NCCL")
def test_abort(self):
id = cuda.nccl.get_unique_id()
comm = cuda.nccl.NcclCommunicator(1, id, 0)
comm.abort()
@attr.gpu
- @unittest.skipUnless(cuda.nccl.get_version() >= 2400, "Using old NCCL")
+ @unittest.skipUnless(cuda.nccl_enabled and
+ cuda.nccl.get_version() >= 2400, "Using old NCCL")
def test_check_async_error(self):
id = cuda.nccl.get_unique_id()
comm = cuda.nccl.NcclCommunicator(1, id, 0)
