Add missing event management to RNG.uniform

inducer · Aug 9, 2016 · 44f011e · 44f011e
1 parent 33245c9
commit 44f011e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pyopencl/clrandom.py b/pyopencl/clrandom.py
@@ -342,7 +342,8 @@ def uniform(self, *args, **kwargs):
 
         result = cl_array.empty(*args, **kwargs)
 
-        self.fill_uniform(result, queue=result.queue, a=a, b=b)
+        result.add_event(
+                self.fill_uniform(result, queue=result.queue, a=a, b=b))
         return result
 
     def fill_normal(self, ary, mu=0, sigma=1, queue=None):
@@ -657,7 +658,8 @@ def uniform(self, *args, **kwargs):
 
         result = cl_array.empty(*args, **kwargs)
 
-        self.fill_uniform(result, queue=result.queue, a=a, b=b)
+        result.add_event(
+                self.fill_uniform(result, queue=result.queue, a=a, b=b))
         return result
 
     def fill_normal(self, ary, mu=0, sigma=1, queue=None):