Skip to main content

dit_block

cortiq_engine::gpu

Function dit_block 

Source 
pub fn dit_block(
    model: &Arc<CmfModel>,
    a: &DitBlockArgs<'_>,
    x: &mut [f32],
) -> bool
Expand description

One whole modulated DiT block on the device — norms, qkv, RoPE, attention, residuals and the SwiGLU FFN in a single command buffer; only x crosses the CPU boundary (in and out).