产品
NVGEMM
NVIDIA提供的GEMM优化组件,将CuTeDSL生成的CUTLASS内核集成到PyTorch Inductor编译后端,用于提升深度学习框架中矩阵乘法的执行性能
时间轴 (近 90 天)
9月11日
PyTorch 2.14 中,NVGEMM 将由 CuTeDSL 生成的 CUTLASS 内核带入了 Inductor 编译后端
待验证50%
NVIDIA提供的GEMM优化组件,将CuTeDSL生成的CUTLASS内核集成到PyTorch Inductor编译后端,用于提升深度学习框架中矩阵乘法的执行性能
PyTorch 2.14 中,NVGEMM 将由 CuTeDSL 生成的 CUTLASS 内核带入了 Inductor 编译后端