Changeset 6f89000 in sasmodels for sasmodels/kernelcl.py

Timestamp:

Sep 24, 2016 5:51:20 PM (8 years ago)

Author:

Paul Kienzle <pkienzle@…>

Children:

d3e3c46a

Parents:

52ec91e

Message:

Allow other processes to run during long computations

File:

• sasmodels/kernelcl.py (modified) (2 diffs)

sasmodels/kernelcl.py

@@ -54 +54 @@
 import warnings
 import logging
+import time
 
 import numpy as np  # type: ignore
@@ -557 +558 @@
         #call_details.show(values)
         # Call kernel and retrieve results
-        last_call = None
-        step = 100
+        wait_for = None
+        last_nap = time.clock()
+        step = 1000000//self.q_input.nq + 1
         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)]
+            wait_for = [kernel(self.queue, self.q_input.global_size, None,
+                               *args, wait_for=wait_for)]
+            if stop < call_details.num_eval:
+                # Allow other processes to run
+                wait_for[0].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)