increase matrix size to show overlap

leofang · leofang · commit 4f861c23fc08 · 2019-10-30T13:55:17.000-04:00
diff --git a/examples/stream/map_reduce.py b/examples/stream/map_reduce.py
@@ -19,8 +19,8 @@
 # Map
 for stream in map_streams:
     with stream:
-        x = rand.normal(size=(1, 1024 * 256))
-        y = rand.normal(size=(1024 * 256, 1))
+        x = rand.normal(size=(1, 1024**2))
+        y = rand.normal(size=(1024**2, 1))
         z = cupy.matmul(x, y)
         zs.append(z)
     stop_event = stream.record()
