Der AI-CUDA-Engineer-Archive-Datensatz ist eine Sammlung von durch KI generierten CUDA-Kerneln, die das anschließende Training von Open-Source-Modellen und die Entwicklung besserer CUDA-Funktionsmodule erleichtern soll. Der Datensatz enthält mehr als 30.000 CUDA-Kernel, die alle von KI-gesteuerten CUDA-Ingenieuren generiert wurden. Bei mehr als 17.000 Kerneln wurde die Richtigkeit verifiziert, und etwa 50% der Kernel übertreffen die native Laufzeit von PyTorch. Es umfasst die PyTorch-Referenzimplementierung, PyTorch/NCU/Clang-tidy-Leistungsanalysedaten, mehrere Kernelversionen für jede Aufgabe, Fehlerinformationen und Leistungsvergleichsdaten mit nativen und kompilierten PyTorch-Laufzeiten.

Zitat

@article{lange2025aicudaengineer, title={Der KI-CUDA-Ingenieur: Agentenbasierte CUDA-Kernel-Erkennung, -Optimierung und -Komposition}, Autor={Lange, Robert Tjarko und Prasad, Aaditya und Sun, Qi und Faldor, Maxence und Tang, Yujin und Ha, David}, Journal={arXiv Preprint}, Jahr={2025} }