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

ep_signal_completion

ep_signal_completion[p2p_world_size: Int, //, use_shmem: Bool, n_experts_per_device: Int = 0, skip_a2a: Bool = False](my_rank: Int32, dst_rank: Int32, recv_count_ptrs: InlineArray[UnsafePointer[UInt64, MutExternalOrigin], p2p_world_size], signal_offset: Int32, signal: UInt64, rank_completion_counter: UnsafePointer[Int32, MutExternalOrigin])

Signals the completion of the communication by writing to the receive count buffer. Will use direct memory access if the target device is on the same node, and use the SHMEM API if the target device is on a different node.

For same-node signaling, uses normal stores and only issues a release store when the last expert for a destination rank is completed. This reduces the number of release stores from n_experts to p2p_world_size.

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