Changeset b297ba9 in sasmodels for sasmodels/kernelcuda.py

Timestamp:

Mar 20, 2019 7:03:50 PM (5 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

master, core_shell_microgels, magnetic_model, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests

Children:

4e28511

Parents:

0d362b7

Message:

lint

File:

• sasmodels/kernelcuda.py (modified) (5 diffs)

sasmodels/kernelcuda.py

@@ -59 +59 @@
 
 import os
-import warnings
 import logging
 import time
@@ -110 +109 @@
 
 def use_cuda():
+    # type: None -> bool
+    """Returns True if CUDA is the default compute engine."""
     sas_opencl = os.environ.get("SAS_OPENCL", "CUDA").lower()
     return HAVE_CUDA and sas_opencl.startswith("cuda")
@@ -136 +137 @@
         if not HAVE_CUDA:
             raise RuntimeError("CUDA startup failed with ***"
-                            + CUDA_ERROR + "***; using C compiler instead")
+                               + CUDA_ERROR + "***; using C compiler instead")
         reset_environment()
         if ENV is None:
@@ -198 +199 @@
     """
     for match in FUNCTION_PATTERN.finditer(source):
-        print(match.group('qualifiers').replace('\n',r'\n'), match.group('function'), '(')
+        print(match.group('qualifiers').replace('\n', r'\n'),
+              match.group('function'), '(')
     return source
 
@@ -260 +262 @@
 
     def release(self):
+        """Free the CUDA device associated with this context."""
         if self.context is not None:
             self.context.pop()