Skip to main content

sample_gemm_data

Function sample_gemm_data 

Source
pub fn sample_gemm_data(m: usize, k: usize, n: usize) -> (Vec<i32>, Vec<i32>)
Expand description

A reproducible int8 (weights K×N row-major, acts M×K row-major) sample in [-128, 127], shared by compile_to_cuda_gemm and the lock so the emitted kernel and the eval_systolic oracle read identical data (no drift). Deterministic in the indices — no RNG.