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_clc_barriers
def init_clc_barriers[num_clc_stages: Int](clc_full_ptr: Pointer[SharedMemBarrier, MutUntrackedOrigin, address_space=AddressSpace.SHARED, _safe=False], clc_empty_ptr: Pointer[SharedMemBarrier, MutUntrackedOrigin, address_space=AddressSpace.SHARED, _safe=False], clc_producer_arv_count: Int32, clc_consumer_arv_count: Int32)
Initialize CLC full/empty barrier pairs.
Called inside the elect_one_warp && elect_one_thread guard for CLC-enabled kernels (default, block_scaled, blockwise_fp8, grouped_2sm).
Parameters:
- βnum_clc_stages (
Int): Number of CLC pipeline stages to initialize barriers for.
Args:
- βclc_full_ptr (
Pointer[SharedMemBarrier, MutUntrackedOrigin, address_space=AddressSpace.SHARED, _safe=False]): Shared-memory pointer to the CLC full barriers, one per pipeline stage. - βclc_empty_ptr (
Pointer[SharedMemBarrier, MutUntrackedOrigin, address_space=AddressSpace.SHARED, _safe=False]): Shared-memory pointer to the CLC empty barriers, one per pipeline stage. - βclc_producer_arv_count (
Int32): Number of arrivals the producer makes on each CLC full barrier. - βclc_consumer_arv_count (
Int32): Number of arrivals the consumer makes on each CLC empty barrier.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!