Adapting Language Models for Low-Resource GPU Kernel Programming

Published: 14 May 2026
on channel: AMD Developer Central
230
8

Speakers: Natalia Pahlavan & Laasya Konidala, Stanford University
Talk Abstract: Large Language Models perform well on high-resource programming languages, but they struggle to generate low-resource, compiler-verified languages such as AMD HIP, where open-source training data is scarce and performance constraints are strict. In this work, we investigate improving HIP kernel generation via (1) synthetic generation of 400 PyTorch tasks with full prompt histories across iterative attempts (2) multi-agent translation and optimization into HIP kernels using evolutionary search and MI350X benchmarking, and (3) SFT on the synthetic dataset followed by GRPO-based RL with hardware-in-the-loop rewards. We evaluate on KernelBench across all three levels using compilation success, functional correctness, and execution speedup relative to PyTorch baselines on the MI350X.

Find the resources you need to develop using AMD products: https://www.amd.com/en/developer.html

Join the Developer Community: https://devcommunity.amd.com/

Join the Developer Discord server:   / discord  

***

© 2026 Advanced Micro Devices, Inc. All rights reserved. AMD, the AMD Arrow logo, EPYC, ROCm, and AMD Instinct and combinations thereof are trademarks of Advanced Micro Devices, Inc.


On this page of the site you can watch the video online Adapting Language Models for Low-Resource GPU Kernel Programming with a duration of hours minute second in good quality, which was uploaded by the user AMD Developer Central 14 May 2026, share the link with friends and acquaintances, this video has already been watched 230 times on youtube and it was liked by 8 viewers. Enjoy your viewing!