forked from huawei/mindspore2022
fix wrong arg call
This commit is contained in:
parent
5afcefdce4
commit
2cf2a67dfd
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue