Changeset 407bf48 in sasmodels for sasmodels/core.py

Timestamp:

Aug 24, 2016 5:19:52 AM (8 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

File:

• sasmodels/core.py (modified) (4 diffs)

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"