flashinfer.comm.PcieIpcReduceScatterLaunchConfig

class flashinfer.comm.PcieIpcReduceScatterLaunchConfig(blocks: int, threads: int, variant: flashinfer.comm.pcie_ipc_collectives.reduce_scatter_policy.PcieIpcReduceScatterVariant)

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

Methods

__init__(blocks, threads, variant)

Attributes

blocks
threads
variant