כתבה
arXiv cs.CL ·
AMDKernelVault: מאגר נתונים ואימון אגנטים
AMDKernelVault: Large-Scale Datasets and Agentic Training for AMD GPU Kernel Optimization
AMDKernelVault הוא מאגר נתונים וכלי אימון לאופטימיזציה של ליבות AMD GPU. הוא כולל 62,153 דוגמאות ליבות HIP ו-2,377 כניסות QA של ספריות ROCm. המאגר שימש לאימון מודל Qwen3-8B.
תקציר מקורי באנגליתarXiv:2609.12471v1 Announce Type: new Abstract: 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-grounded ROCm Libraries QA entries, and 39,893 Triton kernels. We further train Qwen3-8B with supervised fine-tuning and execution-aware reinforcement learnin
קרא במקור המקורי
arxiv.org
פתח כתבה מקורית