Changeset 250fa25 in sasmodels

Timestamp:

Feb 16, 2015 7:01:09 AM (9 years ago)

Author:

pkienzle

Branches:

master, core_shell_microgels, costrafo411, magnetic_model, release_v0.94, release_v0.95, ticket-1257-vesicle-product, ticket_1156, ticket_1265_superball, ticket_822_more_unit_tests

Children:

10576d1

Parents:

f786ff3

Message:

autoselect gpu for opencl if available

Location:

sasmodels

Files:

• kernelcl.py (modified) (3 diffs)
• sasview_model.py (modified) (1 diff)

sasmodels/kernelcl.py

@@ -23 +23 @@
 factors and have instrumental resolution effects applied.
 """
+import os
+import warnings
+
 import numpy as np
 
-import pyopencl as cl
+try:
+    import pyopencl as cl
+except ImportError,exc:
+    warnings.warn(str(exc))
+    raise RuntimeError("OpenCL not available")
+
 from pyopencl import mem_flags as mf
 
@@ -140 +148 @@
     """
     def __init__(self):
-        self.context = cl.create_some_context()
-        self.queues = [cl.CommandQueue(self.context, d)
-                       for d in self.context.devices]
+        # find gpu context
+        #self.context = cl.create_some_context()
+
+        self.context = None
+        if 'PYOPENCL_CTX' in os.environ:
+            self._create_some_context()
+
+        if not self.context:
+            self.context = self._find_context()
+
         # Byte boundary for data alignment
         #self.data_boundary = max(d.min_data_type_align_size
         #                         for d in self.context.devices)
+        self.queues = [cl.CommandQueue(self.context, d)
+                       for d in self.context.devices]
         self.has_double = all(has_double(d) for d in self.context.devices)
         self.compiled = {}
+
+    def _create_some_context(self):
+        try:
+            self.context = cl.create_some_context(interactive=False)
+        except Exception,exc:
+            warnings.warn(str(exc))
+            warnings.warn("pyopencl.create_some_context() failed")
+            warnings.warn("the environment variable 'PYOPENCL_CTX' might not be set correctly")
+
+    def _find_context(self):
+        default = None
+        for platform in cl.get_platforms():
+            for device in platform.get_devices():
+                if device.type == cl.device_type.GPU:
+                    return cl.Context([device])
+                if default is None:
+                    default = device
+
+        if not default:
+            raise RuntimeError("OpenCL device not found")
+
+        return cl.Context([default])
 
     def compile_program(self, name, source, dtype):
@@ -159 +198 @@
             self.compiled[name].release()
             del self.compiled[name]
+
 
 class GpuModel(object):

sasmodels/sasview_model.py

@@ -28 +28 @@
 except ImportError,exc:
     warnings.warn(str(exc))
-    warnings.warn("OpenCL not available --- using ctypes instead")
+    warnings.warn("using ctypes instead")
     from .kerneldll import load_model