https://github.com/NVIDIA/cutlass
- 总 Star 数量
- 10.1k
- 最近更新时间
- 2026-07-27
- 开源时间
- 2017-11-30
项目描述
CUTLASS是NVIDIA开源的CUDA C++模板库,专注于高性能线性代数运算,尤其是矩阵乘法(GEMM)及相关计算。它通过层次化分解和数据移动策略,支持FP64、FP32、TF32、FP16、BF16、块缩放等多种数据类型,并针对Volta至Blackwell等GPU架构进行深度优化。4.6版新增Python DSL(CuTe DSL),允许开发者用Python编写接近硬件的CUDA内核,大幅降低学习曲线与编译时间。适用于深度学习训练、科学计算等需要极致运算性能的场景。