Browse Source

fix wrong arg call

tags/v0.3.0-alpha
shijianning 5 years ago
parent
commit
2cf2a67dfd
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/_akg/add_path.py

+ 2
- 1
mindspore/_akg/add_path.py View File

@@ -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…
Cancel
Save