File tree Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Original file line number Diff line number Diff 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"""
334333function 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" ,
You can’t perform that action at this time.
0 commit comments