可以的,使用triton的话会自动对比选择性能好的triton gemm去进行相关的计算
请问下各位,在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)更高效的矩阵乘法吗?

暂无数据