
LeetCUDA
xlite-dev
📚LeetCUDA: Modern CUDA Learn Notes with PyTorch for Beginners🐑, 200+ CUDA Kernels, Tensor Cores, HGEMM, FA-2 MMA.🎉
AI 简介
LeetCUDA 是一个面向初学者的现代 CUDA 学习资源库,提供结构化学习笔记与可运行示例。核心包含 200+ 经典及前沿 CUDA 核函数(涵盖 Tensor Core 加速、HGEMM 高性能矩阵乘、Flash Attention 2 的 MMA 实现)、对 TF32/F16/BF16/F8 等精度的系统支持,并深度集成 PyTorch 接口。项目强调工程实践性与性能对标(如 HGEMM 接近 cuBLAS 98%~100% TFLOPS),适合 GPU 加速算法研发、大模型推理优化、高性能计算教学及 CUDA 入门训练等场景。
Cuda
GNU General Public License v3.011.5k
Stars
1.2k
Forks
55
Watchers
1
Issues
Star 增长
今日0
近 7 天0
近 30 天+149
综合评分79.23
默认分支main