[LANG] Added seeded random number generation - philox (#261)

2021-09-02 22:02:40 -07:00
parent c069ef907e
commit 8bedcce9be
16 changed files with 595 additions and 6 deletions
--- a/python/tutorials/01-vector-add.py
+++ b/python/tutorials/01-vector-add.py
@@ -43,7 +43,7 @@ def add_kernel(
    y = tl.load(y_ptr + offsets, mask=mask)
    output = x + y
    # Write x + y back to DRAM
-    tl.store(output_ptr + offsets, output)
+    tl.store(output_ptr + offsets, output, mask=mask)


 # %%