Ticket #363: large_kernel.patch

sasmodels/kernelcl.py

@@ -53 +53 @@
 import os
 import warnings
 import logging
+import time
 
 import numpy as np  # type: ignore
 
@@ -556 +557 @@
         #print("Calling OpenCL")
         #call_details.show(values)
         # Call kernel and retrieve results
-        last_call = None
-        step = 100
+        in_process = None
+        last_nap = time.clock()
+        step = 50
         for start in range(0, call_details.num_eval, step):
             stop = min(start + step, call_details.num_eval)
             #print("queuing",start,stop)
             args[1:3] = [np.int32(start), np.int32(stop)]
-            last_call = [kernel(self.queue, self.q_input.global_size,
-                                None, *args, wait_for=last_call)]
+            call = kernel(self.queue, self.q_input.global_size, None,
+                          *args, wait_for=in_process)
+            if stop < call_details.num_eval:
+                # Allow other processes to run
+                # TODO: maybe check time before sleep
+                in_process = [call]
+                call.wait()
+                current_time = time.clock()
+                if current_time - last_nap > 0.5:
+                    time.sleep(0.05)
+                    last_nap = current_time
         cl.enqueue_copy(self.queue, self.result, self.result_b)
         #print("result", self.result)