Fix call mkl gemm in mkldnn.py (#7007)

Co-authored-by: zhangfucheng <zhangfucheng.jason@bytedance.com>
apache · Dec 1, 2020 · fe4c66b · fe4c66b
1 parent 73a1a9a
commit fe4c66b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/contrib/mkldnn.py b/python/tvm/contrib/mkldnn.py
@@ -45,7 +45,7 @@ def matmul(lhs, rhs, transa=False, transb=False, **kwargs):
         (n, m),
         [lhs, rhs],
         lambda ins, outs: tvm.tir.call_packed(
-            "tvm.contrib.mkl.matmul", ins[0], ins[1], outs[0], transa, transb
+            "tvm.contrib.mkldnn.matmul", ins[0], ins[1], outs[0], transa, transb
         ),
         name="C",
         **kwargs,