Skip to content
Snippets Groups Projects
Commit 9975aa8c authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Expose RNG uniform range bounds in more places.

parent 08efb35e
No related branches found
No related tags found
No related merge requests found
...@@ -426,12 +426,12 @@ Generating Arrays of Random Numbers ...@@ -426,12 +426,12 @@ Generating Arrays of Random Numbers
the generator a differing number of times. This function the generator a differing number of times. This function
ensures efficiency. 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` 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). Fill *result* with random values of `dtype` in the range [0,1).
......
...@@ -222,23 +222,23 @@ def _get_generator(queue, luxury=None): ...@@ -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: if queue is None:
queue = result.queue queue = result.queue
gen = _get_generator(queue, luxury=luxury) gen = _get_generator(queue, luxury=luxury)
gen.fill_uniform(result) gen.fill_uniform(result, a=a, b=b)
def rand(*args, **kwargs): 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 from pyopencl.array import Array
luxury = kwargs.pop("luxury", None) luxury = kwargs.pop("luxury", None)
gen = _get_generator(queue, luxury) gen = _get_generator(queue, luxury)
result = Array(queue, shape, dtype) result = Array(queue, shape, dtype)
gen.fill_uniform(result) gen.fill_uniform(result, a=a, b=b)
return result return result
if isinstance(args[0], cl.Context): if isinstance(args[0], cl.Context):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment