Skip to main content

emit_tiled_dp4a_program

logicaffeine_compile::codegen_cuda::gemm_tile

Function emit_tiled_dp4a_program 

Source 
pub fn emit_tiled_dp4a_program(
    tile: &CudaTile,
    m: usize,
    k: usize,
    n: usize,
) -> String
Expand description

Emit a COMPLETE self-contained tiled-DP4A program for shape M×K×N: the tiled kernel plus a host main that fills A/B deterministically, launches gemm_tiled_dp4a<<<(N/BN, M/BM), NTHREAD>>>, and prints the int32 result row-major (one value per line) — so its output is directly comparable to a scalar oracle. Requires the shape to be tile-aligned (M%bm==0, N%bn==0, K%bk==0).