For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo function

init_core_barriers

init_core_barriers[num_input_stages: Int, num_accum_stages: Int](input_barriers_ptr: UnsafePointer[SharedMemBarrier, MutAnyOrigin, address_space=AddressSpace.SHARED], input_consumer_count: Int32, accum_barriers_ptr: UnsafePointer[SharedMemBarrier, MutAnyOrigin, address_space=AddressSpace.SHARED], accum_producer_arv_count: Int32, accum_consumer_arv_count: Int32, tmem_dealloc_ptr: UnsafePointer[SharedMemBarrier, MutAnyOrigin, address_space=AddressSpace.SHARED], tmem_dealloc_thread_count: Int32)

Initialize input, output, and TMEM deallocation barriers.

Called inside the elect_one_warp && elect_one_thread guard. Handles the three barrier init steps shared by all SM100 kernels.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit