pass along verbosity to get_gpu_info related functions

Author: pseudotensor

src/interface_py/h2o4gpu/libs/lib_utils.py

@@ -11,7 +11,7 @@ def __init__(self):
         pass
 
     @staticmethod
-    def get():
+    def get(verbose=1):
         """Get the CPU module object"""
         # SWIG generated files contain some deprecated calls to imp
         with warnings.catch_warnings():
@@ -20,10 +20,11 @@ def get():
                 import h2o4gpu.libs.ch2o4gpu_cpu as ch2o4gpu_cpu
                 return ch2o4gpu_cpu
             except ImportError as e:
-                print("Exception:")
-                print(e)
-                print('\nWarning: h2o4gpu shared object (dynamic library)'
-                      ' for CPU failed to load.')
+                if verbose > 0:
+                    print("Exception:")
+                    print(e)
+                    print('\nWarning: h2o4gpu shared object (dynamic library)'
+                          ' for CPU failed to load.')
                 return None
 
 
@@ -35,7 +36,7 @@ def __init__(self):
         pass
 
     @staticmethod
-    def get():
+    def get(verbose=1):
         """Get the GPU module object"""
         # SWIG generated files contain some deprecated calls to imp
         with warnings.catch_warnings():
@@ -44,17 +45,18 @@ def get():
                 import h2o4gpu.libs.ch2o4gpu_gpu as ch2o4gpu_gpu
                 return ch2o4gpu_gpu
             except ImportError as e:
-                print("Exception:")
-                print(e)
-                print('\nWarning: h2o4gpu shared object (dynamic library)'
-                      ' for GPU failed to load.')
+                if verbose > 0:
+                    print("Exception:")
+                    print(e)
+                    print('\nWarning: h2o4gpu shared object (dynamic library)'
+                          ' for GPU failed to load.')
                 return None
 
 
-def get_lib(n_gpus, devices, verbose=0):
+def get_lib(n_gpus, devices, verbose=1):
     """Load either CPU or GPU H2O4GPU library."""
-    cpu_lib = CPUlib().get()
-    gpu_lib = GPUlib().get()
+    cpu_lib = CPUlib().get(verbose=verbose)
+    gpu_lib = GPUlib().get(verbose=verbose)
     if (n_gpus == 0) or \
             (gpu_lib is None and cpu_lib is not None) or \
             (devices == 0):

src/interface_py/h2o4gpu/util/gpu.py

@@ -178,7 +178,7 @@ def get_gpu_info_c(return_usage=False,
 
     try:
         from ..libs.lib_utils import GPUlib
-        lib = GPUlib().get()
+        lib = GPUlib().get(verbose=0 if not verbose else 1)
 
         total_gpus_actual = \
             lib.get_gpu_info_c(usages_tmp, total_mems_tmp, free_mems_tmp,
@@ -290,8 +290,12 @@ def get_compute_capability(gpu_id):
     """
     Get compute capability for all gpus
     """
-    total_gpus, _, _, majors, minors =\
-        get_gpu_info_c(return_capability=True)
+    try:
+        total_gpus, _, _, majors, minors =\
+            get_gpu_info_c(return_capability=True)
+    # pylint: disable=bare-except
+    except:
+        total_gpus = 0
     if total_gpus > 0:
         gpu_id = gpu_id % total_gpus
         device_major = majors.tolist()[gpu_id]