1
数学库
标题
2024-10-09 10:48:48
71

请问下各位,在DCU K100_AI上使用trition(triton-2.1.0+das1.0+git3841f975.abi0.dtk2404-cp310-cp310
manylinux2014_x86_64.whl)可以做出比torch.matmul(torch-2.1.0+das1.0+git00661e0.abi0.dtk2404-cp310-cp310
manylinux2014_x86_64.whl)更高效的矩阵乘法吗?

 

全部评论(1)
只看楼主
A
ANYA
1楼
可以的,使用triton的话会自动对比选择性能好的triton gemm去进行相关的计算
2024/10/10
没有更多内容了
--
楼主
Level.4
0 帖子
楼主热帖
暂无数据