From e5fea3c8a6c0521212b7bb4c43961816a5bf25fb Mon Sep 17 00:00:00 2001
From: lvwenyuan 00536823 <lvwenyuan@huawei.com>
Date: Thu, 13 Aug 2020 09:40:48 +0800
Subject: [PATCH] change tvm.build with akg.build in composite build_module

---
 python/akg/composite/build_module.py | 4 +++-
 python/akg/ms/message.py             | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/python/akg/composite/build_module.py b/python/akg/composite/build_module.py
index dfa9428..3f725fc 100644
--- a/python/akg/composite/build_module.py
+++ b/python/akg/composite/build_module.py
@@ -15,7 +15,9 @@
 # limitations under the License.
 
 """build module"""
+import os
 import json
+import akg
 from akg import tvm
 from akg.tvm import _api_internal
 from .repository import __all__ as repository
@@ -178,6 +180,6 @@ def build_cuda(outputs, args, sch_name, kernel_name):
         s = scheduler[sch_name](outputs)
         dump_ir = os.getenv('MS_AKG_DUMP_IR') == "on"
         with tvm.build_config(dump_pass_ir = dump_ir):
-            mod = tvm.build(s, args, cuda, name = kernel_name)
+            mod = akg.build(s, list(args), "cuda", name = kernel_name)
             dump_cuda_meta.dump(mod, kernel_name, s, list(args))
             return mod
diff --git a/python/akg/ms/message.py b/python/akg/ms/message.py
index 94e533a..3a856a4 100644
--- a/python/akg/ms/message.py
+++ b/python/akg/ms/message.py
@@ -20,7 +20,7 @@ import json
 import json.decoder as jd
 import logging
 import traceback
-import os.path
+import os
 from pathlib import Path
 
 import akg.tvm
-- 
GitLab