Changeset 750ffa5 in sasmodels for sasmodels/kerneldll.py

Timestamp:

Mar 9, 2015 4:04:55 PM (9 years ago)

Author:

Paul Kienzle <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:

3a45c2c

Parents:

48f0194

Message:

allow test of dll using single precision

File:

• sasmodels/kerneldll.py (modified) (9 diffs)

sasmodels/kerneldll.py

@@ -1 +1 @@
 """
 C types wrapper for sasview models.
+
+The global attribute *ALLOW_SINGLE_PRECISION_DLLS* should be set to *True* if
+you wish to allow single precision floating point evaluation for the compiled
+models, otherwise it defaults to *False*.
 """
+
 import sys
 import os
 import tempfile
 import ctypes as ct
-from ctypes import c_void_p, c_int, c_double
+from ctypes import c_void_p, c_int, c_double, c_float
 
 import numpy as np
@@ -36 +41 @@
 DLL_PATH = tempfile.gettempdir()
 
-
-def dll_path(info):
+ALLOW_SINGLE_PRECISION_DLLS = False
+
+
+def dll_path(info, dtype="double"):
     """
     Path to the compiled model defined by *info*.
@@ -43 +50 @@
     from os.path import join as joinpath, split as splitpath, splitext
     basename = splitext(splitpath(info['filename'])[1])[0]
+    if np.dtype(dtype) == generate.F32:
+        basename += "32"
     return joinpath(DLL_PATH, basename+'.so')
 
 
-def load_model(kernel_module, dtype=None):
+def load_model(kernel_module, dtype="double"):
     """
     Load the compiled model defined by *kernel_module*.
@@ -57 +66 @@
     be defined without using too many resources.
     """
+    if not ALLOW_SINGLE_PRECISION_DLLS: dtype = "double"   # Force 64-bit dll
+    dtype = np.dtype(dtype)
+
     source, info = generate.make(kernel_module)
     if callable(info.get('Iq',None)):
         return PyModel(info)
+
+    if dtype == generate.F32: # 32-bit dll
+        source = generate.use_single(source)
+        tempfile_prefix = 'sas_'+info['name']+'32_'
+    else:
+        tempfile_prefix = 'sas_'+info['name']+'_'
+
     source_files = generate.sources(info) + [info['filename']]
+    dllpath = dll_path(info, dtype)
     newest = max(os.path.getmtime(f) for f in source_files)
-    dllpath = dll_path(info)
     if not os.path.exists(dllpath) or os.path.getmtime(dllpath)<newest:
         # Replace with a proper temp file
-        fid, filename = tempfile.mkstemp(suffix=".c",prefix="sas_"+info['name'])
+        fid, filename = tempfile.mkstemp(suffix=".c",prefix=tempfile_prefix)
         os.fdopen(fid,"w").write(source)
         command = COMPILE%{"source":filename, "output":dllpath}
@@ -76 +95 @@
             #os.unlink(filename); print "saving compiled file in %r"%filename
             pass
-    return DllModel(dllpath, info)
+    return DllModel(dllpath, info, dtype=dtype)
 
 
@@ -96 +115 @@
     Call :meth:`release` when done with the kernel.
     """
-    def __init__(self, dllpath, info):
+    def __init__(self, dllpath, info, dtype=generate.F32):
         self.info = info
         self.dllpath = dllpath
         self.dll = None
+        self.dtype = np.dtype(dtype)
 
     def _load_dll(self):
@@ -110 +130 @@
         self.dll = ct.CDLL(self.dllpath)
 
-        pd_args_1d = [c_void_p, c_double] + [c_int]*Npd1d if Npd1d else []
-        pd_args_2d= [c_void_p, c_double] + [c_int]*Npd2d if Npd2d else []
+        fp = c_float if self.dtype == generate.F32 else c_double
+        pd_args_1d = [c_void_p, fp] + [c_int]*Npd1d if Npd1d else []
+        pd_args_2d= [c_void_p, fp] + [c_int]*Npd2d if Npd2d else []
         self.Iq = self.dll[generate.kernel_name(self.info, False)]
-        self.Iq.argtypes = IQ_ARGS + pd_args_1d + [c_double]*Nfixed1d
+        self.Iq.argtypes = IQ_ARGS + pd_args_1d + [fp]*Nfixed1d
 
         self.Iqxy = self.dll[generate.kernel_name(self.info, True)]
-        self.Iqxy.argtypes = IQXY_ARGS + pd_args_2d + [c_double]*Nfixed2d
+        self.Iqxy.argtypes = IQXY_ARGS + pd_args_2d + [fp]*Nfixed2d
 
     def __getstate__(self):
@@ -125 +146 @@
 
     def __call__(self, q_input):
+        if self.dtype != q_input.dtype:
+            raise TypeError("data is %s kernel is %s" % (q_input.dtype, self.dtype))
         if self.dll is None: self._load_dll()
         kernel = self.Iqxy if q_input.is_2D else self.Iq
@@ -138 +161 @@
         ctypes and some may be pure python.
         """
-        return PyInput(q_vectors, dtype=F64)
+        return PyInput(q_vectors, dtype=self.dtype)
 
     def release(self):