ENH: Updated cl.Context init arguement device_type= to allow a string.

srossross · Jan 10, 2012 · 90f3ddf · 90f3ddf
1 parent 036f009
commit 90f3ddf
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/opencl/copencl.pyx b/opencl/copencl.pyx
@@ -177,11 +177,20 @@ cdef class Device:
     processors such as DSPs and the Cell/B.E. processor.
     
     '''
-    DEFAULT = CL_DEVICE_TYPE_DEFAULT
+    ACCELERATOR = CL_DEVICE_TYPE_ACCELERATOR
     ALL = CL_DEVICE_TYPE_ALL
     CPU = CL_DEVICE_TYPE_CPU
+    DEFAULT = CL_DEVICE_TYPE_DEFAULT
     GPU = CL_DEVICE_TYPE_GPU
 
+    DEV_TYPE_MAP = {
+        'ACCELERATOR' : CL_DEVICE_TYPE_ACCELERATOR,
+        'ALL' : CL_DEVICE_TYPE_ALL,
+        'CPU' : CL_DEVICE_TYPE_CPU,
+        'DEFAULT' : CL_DEVICE_TYPE_DEFAULT,
+        'GPU' : CL_DEVICE_TYPE_GPU,
+                   }
+
     cdef cl_device_id device_id
 
     def __cinit__(self):