From 2cf2a67dfda485307d3a2b739df5ab107ccde7a5 Mon Sep 17 00:00:00 2001 From: shijianning Date: Wed, 13 May 2020 16:21:24 +0800 Subject: [PATCH] fix wrong arg call --- mindspore/_akg/add_path.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/_akg/add_path.py b/mindspore/_akg/add_path.py index 2de2f73e1c..d1e50f8177 100644 --- a/mindspore/_akg/add_path.py +++ b/mindspore/_akg/add_path.py @@ -31,8 +31,9 @@ def AKGAddPath(): class AKGMetaPathFinder: """class AKGMetaPath finder.""" - def find_module(self, fullname): + def find_module(self, fullname, path=None): """method _akg find module.""" + _ = path if fullname.startswith("_akg.tvm"): rname = fullname[5:] return AKGMetaPathLoader(rname)