Skip to content

Commit 4e09fcd

Browse files
authored
fix gpu yield test (#49)
1 parent 9ba3723 commit 4e09fcd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_gpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -424,7 +424,7 @@ def reduced(lhs: T.i32(), rhs: T.i32()):
424424
%0 = memref.load %alloc[%arg0, %arg3] : memref<2x6xi32>
425425
%1 = gpu.all_reduce %0 uniform {
426426
^bb0(%arg12: i32, %arg13: i32):
427-
"gpu.yield"(%arg12) : (i32) -> ()
427+
gpu.yield %arg12 : i32
428428
} : (i32) -> i32
429429
memref.store %1, %alloc_0[%arg0] : memref<2xi32>
430430
gpu.terminator

0 commit comments

Comments
 (0)