Skip to main content

shard

logicaffeine_compile::spatial::passes::shard

Function shard 

Source 
pub fn shard(
    p: &SpatialProgram,
    plan: &ShardPlan,
    target: &SpatialTarget,
) -> Result<SpatialProgram, ShardError>
Expand description

Split p over plan.chips RDUs along the contraction axis, insert the all-reduce, and re-run the pinned passes so the collective is scheduled and certified like any other hand-off.