Changeset 407bf48 in sasmodels

Timestamp:

Aug 24, 2016 3:19:52 AM (9 years ago)

Author:

wojciech

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:

8407d8c

Parents:

0dc34c3

Message:

gpu flag added to sasmodels

Location:

sasmodels

Files:

• compare.py (modified) (1 diff)
• core.py (modified) (4 diffs)
• modelinfo.py (modified) (1 diff)

sasmodels/compare.py

@@ -563 +563 @@
     elif dtype.endswith('!'):
         return eval_ctypes(model_info, data, dtype=dtype[:-1], cutoff=cutoff)
+    elif not model_info.gpu:
+        return eval_ctypes(model_info, data, dtype=dtype, cutoff=cutoff)
     else:
         return eval_opencl(model_info, data, dtype=dtype, cutoff=cutoff)

sasmodels/core.py

@@ -46 +46 @@
 #    build_model
 
-KINDS = ("all", "py", "c", "double", "single", "1d", "2d",
+KINDS = ("all", "py", "c", "double", "single", "gpu", "1d", "2d",
          "nonmagnetic", "magnetic")
 def list_models(kind=None):
@@ -60 +60 @@
         * single: models which support single precision
         * double: models which require double precision
+        * gpu: models which compiles on GPU
         * 1d: models which are 1D only, or 2D using abs(q)
         * 2d: models which can be 2D
@@ -85 +86 @@
         return True
     elif kind == "single" and info.single:
+        return True
+    elif kind == "gpu" and info.gpu:
         return True
     elif kind == "2d" and any(p.type == 'orientation' for p in pars):
@@ -215 +218 @@
     """
     # Assign default platform, overriding ocl with dll if OpenCL is unavailable
+    # If gpu=False OpenCL is switched off
     if platform is None:
         platform = "ocl"
-    if platform == "ocl" and not HAVE_OPENCL:
+    if platform == "ocl" and not HAVE_OPENCL or not model_info.gpu:
         platform = "dll"
 

sasmodels/modelinfo.py

@@ -731 +731 @@
     info.category = getattr(kernel_module, 'category', None)
     info.single = getattr(kernel_module, 'single', True)
+    info.gpu = getattr(kernel_module, 'gpu', True)
     info.structure_factor = getattr(kernel_module, 'structure_factor', False)
     info.profile_axes = getattr(kernel_module, 'profile_axes', ['x', 'y'])