calling is_nvls_supported function

microsoft · Nov 22, 2024 · 2e958d9 · 2e958d9
1 parent 2a1eda9
commit 2e958d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/test/executor_test.py b/python/test/executor_test.py
@@ -130,7 +130,7 @@ def dtype_to_mscclpp_dtype(dtype):
 
 
 def allocate_buffer(nelems, dtype):
-    if is_nvls_supported:
+    if is_nvls_supported():
         buffer_raw = alloc_shared_physical_cuda(nelems * cp.dtype(dtype).itemsize)
         buffer_ptr = cp.cuda.MemoryPointer(
             cp.cuda.UnownedMemory(buffer_raw.get_ptr(), buffer_raw.size(), buffer_raw), 0