Skip to content

Commit 35dc512

Browse files
Regenerate MLIR Bindings (#1830)
Co-authored-by: enzyme-ci-bot[bot] <78882869+enzyme-ci-bot[bot]@users.noreply.github.com>
1 parent b614095 commit 35dc512

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

src/mlir/Dialects/MosaicGPU.jl

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -325,20 +325,22 @@ end
325325
"""
326326
`initialize_barrier`
327327
328-
Initializes a memref of barriers each meant to synchronize exactly
328+
Initializes `num_barriers` barriers each meant to synchronize exactly
329329
`arrival_count` threads.
330330
331-
The base pointer of the result memref corresponds to `base_pointer`, which
332-
must be a pointer to a shared memory location.
331+
`base_pointer` must be a pointer to a shared memory location.
333332
"""
334333
function initialize_barrier(
335-
base_pointer::Value; barriers_ref::IR.Type, arrival_count, location=Location()
334+
base_pointer::Value; arrival_count, num_barriers, location=Location()
336335
)
337-
op_ty_results = IR.Type[barriers_ref,]
336+
op_ty_results = IR.Type[]
338337
operands = Value[base_pointer,]
339338
owned_regions = Region[]
340339
successors = Block[]
341-
attributes = NamedAttribute[namedattribute("arrival_count", arrival_count),]
340+
attributes = NamedAttribute[
341+
namedattribute("arrival_count", arrival_count),
342+
namedattribute("num_barriers", num_barriers),
343+
]
342344

343345
return create_operation(
344346
"mosaic_gpu.initialize_barrier",

0 commit comments

Comments
 (0)