LV.2
#1 tanhaojue回复
您好,需要在路径:/torch/src/catch/torch_mlu/tools/template/jit/mlu_supported_ops.h下添加matmul算子,之后重新编译cambricon-pytorch,查看编译后生成的融合白名单(路径为:/torch/src/catch/torch_mlu/csrc/jit/passes/mlu_supported_ops.h),可看到matmul算子已被加入。
2021-08-31 14:38:11
0|
回复