tvm_tirx.tirx.script.cpu_parallel_barrier

tvm_tirx.tirx.script.cpu_parallel_barrier#

tvm_tirx.tirx.script.cpu_parallel_barrier(*, ty=None, loc=UNKNOWN_LOC)#

Synchronize all workers in the current CPU parallel launch.

Every worker must reach this operation at the same program point. Place it inside a parallel_launch region, outside parallel loops, whose iteration counts can differ between workers. Writes before the barrier are visible to all workers after it.

To replace pragma_parallel_barrier_when_finish, place this operation after the former attribute body.

Parameters:

loc (Location)