fix

microsoft · Jan 23, 2024 · 8e32bd2 · 8e32bd2
1 parent 2eb20af
commit 8e32bd2
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/python/mscclpp/utils.py b/python/mscclpp/utils.py
@@ -154,5 +154,6 @@ def pack(*args):
             raise RuntimeError(f"Unsupported type: {type(arg)}")
     return res
 
+
 def is_nvls_supported():
     return cp.cuda.runtime.runtimeGetVersion() >= 12010
diff --git a/src/nvls_connection.cc b/src/nvls_connection.cc
@@ -4,6 +4,7 @@
 #include <sys/syscall.h>
 #include <unistd.h>
 
+#include <algorithm>
 #include <mscclpp/core.hpp>
 #include <mscclpp/utils.hpp>
 

diff --git a/src/utils.cc b/src/utils.cc
@@ -70,10 +70,10 @@ std::string getHostName(int maxlen, const char delim) {
 bool isNvlsSupported() {
 #if (CUDART_VERSION >= 12010)
   CUdevice dev;
-  int nvlsSupport;
+  int isNvlsSupported;
   MSCCLPP_CUTHROW(cuCtxGetDevice(&dev));
-  MSCCLPP_CUTHROW(cuDeviceGetAttribute(&nvlsSupport, CU_DEVICE_ATTRIBUTE_MULTICAST_SUPPORTED, dev));
-  return nvlsSupport == 1;
+  MSCCLPP_CUTHROW(cuDeviceGetAttribute(&isNvlsSupported, CU_DEVICE_ATTRIBUTE_MULTICAST_SUPPORTED, dev));
+  return isNvlsSupported == 1;
 #endif
   return false;
 }