flashinfer.comm.PcieIpcAllGatherLaunchConfig

class flashinfer.comm.PcieIpcAllGatherLaunchConfig(blocks: int, threads: int, variant: flashinfer.comm.pcie_ipc_collectives.all_gather_policy.PcieIpcAllGatherVariant)

__init__(blocks: int, threads: int, variant: PcieIpcAllGatherVariant) → None

Methods

__init__(blocks, threads, variant)

Attributes

blocks
threads
variant