SOURCE-LINKED INTELLIGENCE

AMDKernelVault: Large-Scale Datasets and Agentic Training for AMD GPU Kernel Optimization

arXiv · AI, language, vision and robotics · article · Sep 11, 2026 · UTC

We introduce AMDKernelVault, an open HIP and Triton kernel corpus and training framework for recent AMD CDNA GPUs. Existing LLM-based kernel agents are largely CUDA/NVIDIA-centric and often depend on repeated frontier-LLM calls for generation, reflection, and optimization. To address this gap, we develop HIPKernelGen and TritonKernelGen, agent-driven pipelines that transform PyTorch references into HIP or Triton kernels, compile and validate candidates under ROCm, and latency-profile them on AMD hardware. The corpus contains 62,153 execution-verified HIP kernel samples, 2,377 production-ground

Read original source ↗ Open in workspace

recordType

paper

region

Global