We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9ba3723 commit 4e09fcdCopy full SHA for 4e09fcd
tests/test_gpu.py
@@ -424,7 +424,7 @@ def reduced(lhs: T.i32(), rhs: T.i32()):
424
%0 = memref.load %alloc[%arg0, %arg3] : memref<2x6xi32>
425
%1 = gpu.all_reduce %0 uniform {
426
^bb0(%arg12: i32, %arg13: i32):
427
- "gpu.yield"(%arg12) : (i32) -> ()
+ gpu.yield %arg12 : i32
428
} : (i32) -> i32
429
memref.store %1, %alloc_0[%arg0] : memref<2xi32>
430
gpu.terminator
0 commit comments