diff --git a/pystencils/rng.py b/pystencils/rng.py index 26a92b313de13343c02375f6bfb003fca15b418f..91cccaeef11a9f30d29bfc2cae33779877ecca9e 100644 --- a/pystencils/rng.py +++ b/pystencils/rng.py @@ -118,8 +118,12 @@ class PhiloxFourFloats(CustomCodeNode): def random_symbol(assignment_list, rng_node=PhiloxTwoDoubles, *args, **kwargs): + if 'keys' not in kwargs: + kwargs['keys'] = (0, 0) + while True: node = rng_node(*args, **kwargs) + kwargs['keys'] = (kwargs['keys'][0] + 1, kwargs['keys'][1]) inserted = False for symbol in node.result_symbols: if not inserted: