From 9975aa8c0e6c6e80919cad1d5b18ae878170fc4c Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sat, 19 Nov 2011 19:56:55 -0500
Subject: [PATCH] Expose RNG uniform range bounds in more places.
---
doc/source/array.rst | 6 +++---
pyopencl/clrandom.py | 8 ++++----
2 files changed, 7 insertions(+), 7 deletions(-)
diff --git a/doc/source/array.rst b/doc/source/array.rst
index e2003ff8..291228e3 100644
--- a/doc/source/array.rst
+++ b/doc/source/array.rst
@@ -426,12 +426,12 @@ Generating Arrays of Random Numbers
the generator a differing number of times. This function
ensures efficiency.
-.. function:: rand(queue, shape, dtype)
+.. function:: rand(queue, shape, dtype, a=0, b=1)
Return an array of `shape` filled with random values of `dtype`
- in the range [0,1).
+ in the range [a,b).
-.. function:: fill_rand(result, queue=None)
+.. function:: fill_rand(result, queue=None, a=0, b=1)
Fill *result* with random values of `dtype` in the range [0,1).
diff --git a/pyopencl/clrandom.py b/pyopencl/clrandom.py
index d0c2a2a0..c0475854 100644
--- a/pyopencl/clrandom.py
+++ b/pyopencl/clrandom.py
@@ -222,23 +222,23 @@ def _get_generator(queue, luxury=None):
-def fill_rand(result, queue=None, luxury=4):
+def fill_rand(result, queue=None, luxury=4, a=0, b=1):
if queue is None:
queue = result.queue
gen = _get_generator(queue, luxury=luxury)
- gen.fill_uniform(result)
+ gen.fill_uniform(result, a=a, b=b)
def rand(*args, **kwargs):
- def inner_rand(queue, shape, dtype, luxury=None):
+ def inner_rand(queue, shape, dtype, luxury=None, a=0, b=1):
from pyopencl.array import Array
luxury = kwargs.pop("luxury", None)
gen = _get_generator(queue, luxury)
result = Array(queue, shape, dtype)
- gen.fill_uniform(result)
+ gen.fill_uniform(result, a=a, b=b)
return result
if isinstance(args[0], cl.Context):
--
GitLab