[RUNTIME] Bunch of bugfixes (#372)

2021-11-12 00:55:00 -08:00
parent f7ab96cfd7
commit e66bf76354
5 changed files with 128 additions and 17 deletions
--- a/python/triton/language/random.py
+++ b/python/triton/language/random.py
@@ -110,6 +110,7 @@ def randint4x(seed, offset):
    :param offsets: The offsets to generate random numbers for.
    """
    z = offset*0 #FIXME: just 0 doesn't work. Likelye some error with broadcasting
+    seed = seed + 0
    seed = hacky_to_uint64(seed) # uint will solve this
    seed_hi = ((seed >> 32) & 0xffffffff).to(tl.int32)
    seed_lo = (seed & 0xffffffff).to(tl.int32)