fp_contract(fast) for pocl CUDA

Author: isuruf

sumpy/codegen.py

@@ -208,6 +208,20 @@ def register_bessel_callables(loopy_knl):
             Hankel1_01("hank1_01"))
     return loopy_knl
 
+
+def _fp_contract_fast_preamble(preamble_info):
+    yield ("fp_contract_fast_pocl", "#pragma clang fp contract(fast)")
+
+
+def register_optimization_preambles(loopy_knl, device):
+    if isinstance(loopy_knl.target, lp.PyOpenCLTarget):
+        import pyopencl as cl
+        if device.platform.name == "Portable Computing Language" and \
+                (device.type & cl.device_type.GPU):
+            loopy_knl = lp.register_preamble_generators(loopy_knl,
+                [_fp_contract_fast_preamble])
+    return loopy_knl
+
 # }}}
 
 

sumpy/p2p.py

@@ -190,6 +190,9 @@ def get_optimized_kernel(self, targets_is_obj_array, sources_is_obj_array):
         knl = lp.set_options(knl,
                 enforce_variable_access_ordered="no_check")
 
+        from sumpy.codegen import register_optimization_preambles
+        knl = register_optimization_preambles(knl, self.device)
+
         return knl
 
 
@@ -714,6 +717,9 @@ def get_optimized_kernel(self, max_nsources_in_one_box,
         knl = lp.set_options(knl,
                 enforce_variable_access_ordered="no_check")
 
+        from sumpy.codegen import register_optimization_preambles
+        knl = register_optimization_preambles(knl, self.device)
+
         return knl
 
     def __call__(self, queue, **kwargs):