Robust, comprehensive evaluation for AI-driven kernel generation
Develop robust and comprehensive evaluation protocols for AI-driven GPU kernel generation that jointly assess robustness and generalization across input shapes, operator types, and hardware ecosystems, overcoming current limitations of benchmarks confined to fixed shapes and NVIDIA-only forward-pass primitives.
References
A key open challenge in AI-driven kernel generation is the lack of robust and comprehensive evaluation.
Existing RTL-agent benchmarks generally do not include a mechanism to detect over-solving or reward hacking . They measure whether the submitted artifact passes the provided harness, but they usually do not separate debugging feedback from final hidden scoring, test robustness under randomized or perturbed stimuli, or audit whether a solution has specialized to artifacts of the evaluator. This is an open problem for the community.
Several limitations remain, and each is a future direction this work leaves open. i) Section~\ref{sec:theory} says enough about bit-level semantics for a descriptor to hold across machines and across library versions: none of its fields names a machine, and the split is written down rather than read off the hardware, so it survives the change in SM count at which cuBLAS's own guarantee lapses. Measuring that cross-machine robustness is what the resources behind this work did not reach, due to resource limitations.