Towards Automated Kernel Generation in the Era of LLMs
IntermediateYang Yu, Peiyu Zang et al.Jan 22arXiv
AI programs called LLMs can now help write the tiny, super-fast pieces of code (kernels) that make GPUs run AI models efficiently.
#LLM-driven kernel generation#GPU kernels#CUDA