pep8

theano/sandbox/cuda/__init__.py

@@ -23,7 +23,8 @@ if config.cuda.root == "AUTO":
     # set nvcc_path correctly and get the version
     nvcc_compiler.set_cuda_root()
 
-#is_nvcc_available called here to initialize global vars in nvcc_compiler module
+#is_nvcc_available called here to initialize global vars in
+#nvcc_compiler module
 nvcc_compiler.is_nvcc_available()
 
 # Compile cuda_ndarray.cu
@@ -31,8 +32,9 @@ nvcc_compiler.is_nvcc_available()
 # printed and this module will not be working properly (we set `cuda_available`
 # to False).
 
-# This variable is True by default, and set to False if nvcc is not available or
-# their is no cuda card or something goes wrong when trying to initialize cuda.
+# This variable is True by default, and set to False if nvcc is not
+# available or their is no cuda card or something goes wrong when
+# trying to initialize cuda.
 cuda_available = True
 
 # Global variable to avoid displaying the same warning multiple times.
@@ -41,6 +43,7 @@ cuda_warning_is_displayed = False
 #This variable is set to True when we enable cuda.(i.e. when use() is called)
 cuda_enabled = False
 
+
 # Code factorized within a function so that it may be called from multiple
 # places (which is not currently the case, but may be useful in the future).
 def set_cuda_disabled():
@@ -72,17 +75,18 @@ libcuda_ndarray_so = os.path.join(cuda_ndarray_loc,
                                'libcuda_ndarray.' + get_lib_extension())
 
 
-# Add the theano cache directory's cuda_ndarray subdirectory to the list of
-# places that are hard-coded into compiled modules' runtime library search
-# list.  This works in conjunction with nvcc_compiler.nvcc_module_compile_str
-# which adds this folder during compilation with -L and also adds -lcuda_ndarray
-# when compiling modules.
+# Add the theano cache directory's cuda_ndarray subdirectory to the
+# list of places that are hard-coded into compiled modules' runtime
+# library search list.  This works in conjunction with
+# nvcc_compiler.nvcc_module_compile_str which adds this folder during
+# compilation with -L and also adds -lcuda_ndarray when compiling
+# modules.
 nvcc_compiler.add_standard_rpath(cuda_ndarray_loc)
 
 compile_cuda_ndarray = True
 
 if os.path.exists(cuda_ndarray_so):
-    compile_cuda_ndarray = date>=os.stat(cuda_ndarray_so)[stat.ST_MTIME]
+    compile_cuda_ndarray = date >= os.stat(cuda_ndarray_so)[stat.ST_MTIME]
 if not compile_cuda_ndarray:
     try:
         # If we load a previously-compiled version, config.compiledir should
@@ -111,7 +115,7 @@ try:
                     include_dirs=[cuda_path], libs=['cublas'])
             from cuda_ndarray.cuda_ndarray import *
 except Exception, e:
-    _logger.error( "Failed to compile cuda_ndarray.cu: %s", str(e))
+    _logger.error("Failed to compile cuda_ndarray.cu: %s", str(e))
     set_cuda_disabled()
 
 if cuda_available:
@@ -132,7 +136,7 @@ if cuda_available:
         gpu_init()
         cuda_available = True
         cuda_initialization_error_message = ""
-        # actively closing our gpu session presents segfault-on-exit on some systems
+# actively closing our gpu session presents segfault-on-exit on some systems
         atexit.register(gpu_shutdown)
     except EnvironmentError, e:
         cuda_available = False
@@ -199,10 +203,11 @@ def use(device,
     global cuda_enabled, cuda_initialization_error_message
     if force and not cuda_available and device.startswith('gpu'):
         if not nvcc_compiler.is_nvcc_available():
-            raise EnvironmentError("You forced the use of gpu device '%s', but "
-                                   "nvcc was not found. Set it in your PATH "
+            raise EnvironmentError("You forced the use of gpu device '%s', but"
+                                   " nvcc was not found. Set it in your PATH "
                                    "environment variable or set the Theano "
-                                   "flags 'cuda.root' to its directory" % device)
+                                   "flags 'cuda.root' to its directory"
+                                   "" % device)
         else:
             raise EnvironmentError("You forced the use of gpu device %s, "
                                    "but CUDA initialization failed "
@@ -217,7 +222,8 @@ def use(device,
         try:
             if cuda_initialization_error_message:
                 error_addendum = " (error: %s)" % cuda_initialization_error_message
-        except NameError: # cuda_initialization_error_message is not available b/c compilation failed
+        except NameError:
+# cuda_initialization_error_message is not available b/c compilation failed
             pass
         _logger.warning('CUDA is installed, but device %s is not available %s',
                 device, error_addendum)
@@ -233,12 +239,12 @@ def use(device,
         raise ValueError("Invalid device identifier", device)
     if use.device_number is None:
         # No successful call to use() has been made yet
-        if device != 'gpu' and device<0:
+        if device != 'gpu' and device < 0:
             return
-        if device in [None,""]:
-            device=0
+        if device in [None, ""]:
+            device = 0
         try:
-            if device !='gpu':
+            if device != 'gpu':
                 gpu_init(device)
 
             if move_shared_float32_to_gpu:
@@ -247,14 +253,15 @@ def use(device,
 
             if enable_cuda:
                 cuda_enabled = True
-            print >> sys.stderr, "Using gpu device %d: %s" % (active_device_number(), active_device_name())
+            print >> sys.stderr, "Using gpu device %d: %s" % (
+                active_device_number(), active_device_name())
         except (EnvironmentError, ValueError), e:
             _logger.error(("ERROR: Not using GPU."
                 " Initialisation of device %i failed:\n%s"),
                 device, e)
             cuda_enabled = False
             if force:
-                e.args+=(("You asked to force this device and it failed."
+                e.args += (("You asked to force this device and it failed."
                             " No fallback to the cpu or other gpu device."),)
                 raise
 
@@ -275,17 +282,16 @@ def use(device,
         try:
             #in case the device if just gpu,
             # we check that the driver init it correctly.
-            cuda_ndarray.cuda_ndarray.CudaNdarray.zeros((5,5))
+            cuda_ndarray.cuda_ndarray.CudaNdarray.zeros((5, 5))
         except (Exception, NameError), e:
             # NameError when no gpu present as cuda_ndarray is not loaded.
-            e.args+=("ERROR: GPU forced but failed. ",)
+            e.args += ("ERROR: GPU forced but failed. ",)
             raise
-
-
 use.device_number = None
 
+
 def handle_shared_float32(tf):
-    """Set the CudaNdarrayType as the default handler for shared float32 arrays.
+    """Set the default shared type for float32 tensor to CudaNdarrayType
 
     This function is intended to be called from use(gpu_index), not directly.
     """
@@ -299,7 +305,8 @@ def handle_shared_float32(tf):
 if config.device.startswith('gpu'):
     use(device=config.device, force=config.force_device)
 elif config.init_gpu_device:
-    assert config.device=="cpu", ("We can use the Theano flag init_gpu_device"
+    assert config.device == "cpu", (
+        "We can use the Theano flag init_gpu_device"
         " only when the Theano flag device=='cpu'")
     _logger.warning(("GPU device %s will be initialized, and used if a GPU is "
           "needed. "