Set bar1_total = 0 if it can't be determined

rapidsai · Aug 24, 2023 · c9414fc · c9414fc
1 parent 5102d44
commit c9414fc
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ucp/__init__.py b/ucp/__init__.py
@@ -67,8 +67,9 @@ def _is_mig_device(handle):
             try:
                 bar1_total = pynvml.nvmlDeviceGetBAR1MemoryInfo(handle).bar1Total
             except pynvml.nvml.NVMLError_NotSupported:
-                # Bar1 access not supported on this device. Skip
-                continue
+                # Bar1 access not supported on this device, set it to
+                # zero (always lower than device memory).
+                bar1_total = 0
 
             total_memory = pynvml.nvmlDeviceGetMemoryInfo(handle).total
             if total_memory <= bar1_total: