Skip to main content

compile_program_to_spatial_tp

Function compile_program_to_spatial_tp 

Source
pub fn compile_program_to_spatial_tp(
    source: &str,
    target: &SpatialTarget,
    manifest: &TpManifest,
) -> Result<ProgramLowering, SpatialReject>
Expand description

compile_program_to_spatial for a TENSOR-PARALLEL program (see super::tensor_parallel): every section runs on the chip manifest gives what it computes, and each collective’s fold statements are one all-reduce section.