From 427876ec510e3b83a4b72730da8c73e35070ee6d Mon Sep 17 00:00:00 2001 From: Logan Weber Date: Sat, 10 Aug 2019 01:45:28 +0000 Subject: [PATCH 1/2] Revert compile_cmd kwarg name change --- python/tvm/contrib/cc.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/tvm/contrib/cc.py b/python/tvm/contrib/cc.py index 43d653255ded..2c8953fd5485 100644 --- a/python/tvm/contrib/cc.py +++ b/python/tvm/contrib/cc.py @@ -27,7 +27,7 @@ def create_shared(output, objects, options=None, - compile_cmd="g++"): + cc="g++"): """Create shared library. Parameters @@ -41,11 +41,11 @@ def create_shared(output, options : List[str] The list of additional options string. - compile_cmd : Optional[str] + cc : Optional[str] The compiler command. """ if sys.platform == "darwin" or sys.platform.startswith("linux"): - _linux_compile(output, objects, options, compile_cmd) + _linux_compile(output, objects, options, cc) elif sys.platform == "win32": _windows_shared(output, objects, options) else: From 0a5545a5c4119989513d8d5a92dde8311fb746d4 Mon Sep 17 00:00:00 2001 From: Logan Weber Date: Sat, 10 Aug 2019 16:23:11 +0000 Subject: [PATCH 2/2] Fix binutil tests --- tests/python/contrib/test_binutil.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/contrib/test_binutil.py b/tests/python/contrib/test_binutil.py index 617a758d2752..e4ebd74a1a9c 100644 --- a/tests/python/contrib/test_binutil.py +++ b/tests/python/contrib/test_binutil.py @@ -43,7 +43,7 @@ def make_binary(): with open(tmp_source, "w") as f: f.write(prog) cc.create_shared(tmp_obj, tmp_source, [], - compile_cmd="{}gcc".format(TOOLCHAIN_PREFIX)) + cc="{}gcc".format(TOOLCHAIN_PREFIX)) prog_bin = bytearray(open(tmp_obj, "rb").read()) return prog_bin