跳到正文
原文
Hugging Face Blog·· 2025-08-18AI 评分48

从零到 GPU:构建与扩展生产级 CUDA Kernel 指南

From Zero to GPU: A Guide to Building and Scaling Production-Ready CUDA Kernels

AI 导读

Hugging Face 发布 kernel-builder 库,让开发者可在本地开发自定义 CUDA kernel,再为多种架构构建并通过 Hub 分发。指南以 RGB 转灰度 kernel 为例,演示用 PyTorch 现代 C++ API 将函数注册为原生算子,从而兼容 torch.compile 并支持硬件特定实现。

来源:Hugging Face Blog · huggingface.co