[控场AI]
产品

NVGEMM

NVIDIA提供的GEMM优化组件,将CuTeDSL生成的CUTLASS内核集成到PyTorch Inductor编译后端,用于提升深度学习框架中矩阵乘法的执行性能

时间轴 (近 90 天)

9月11日

PyTorch 2.14 中,NVGEMM 将由 CuTeDSL 生成的 CUTLASS 内核带入了 Inductor 编译后端

待验证50%

全部知识事实 (1)

来源文章