From 44f011e59f7c3deb6ac2703be360753f9fcf36e6 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 9 Aug 2016 13:28:14 -0500
Subject: [PATCH] Add missing event management to RNG.uniform
---
pyopencl/clrandom.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/pyopencl/clrandom.py b/pyopencl/clrandom.py
index 98e71b49..225705c2 100644
--- a/pyopencl/clrandom.py
+++ b/pyopencl/clrandom.py
@@ -342,7 +342,8 @@ class RanluxGenerator(object):
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 @@ class Random123GeneratorBase(object):
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):
--
GitLab