Skip to content

Commit 313caec

Browse files
authored
Fix nbarrier error in gen test script (#738)
1 parent 23ee1e9 commit 313caec

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/xetile-test-gen/xetile_testgen.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -254,8 +254,8 @@ def gen_prologue(self):
254254
template += f"""
255255
xegpu.alloc_nbarrier {number_of_dpas_c_tile}
256256
%nbarrier_id = arith.constant 1 : i8
257-
%nbarrier_role = arith.constant 0 : i8
258-
%nbarrier = xegpu.create_nbarrier %nbarrier_id, %nbarrier_role {{num_producers = {self.params.SG.layout.num_elements()} : i8, num_consumers = {self.params.SG.layout.num_elements()} : i8}} : (i8, i8) -> !xegpu.nbarrier
257+
%nthreads = arith.constant {self.params.SG.layout.num_elements()} : i8
258+
%nbarrier = xegpu.init_nbarrier %nbarrier_id, %nthreads : i8, i8 -> !xegpu.nbarrier
259259
"""
260260
return template
261261

0 commit comments

Comments
 (0)