From af090e4a8a3521f3a4e26ba719408bb52577f299 Mon Sep 17 00:00:00 2001 From: Logan Weber <36520469+weberlo@users.noreply.github.com> Date: Mon, 12 Aug 2019 08:52:19 -0700 Subject: [PATCH] Revert compile_cmd kwarg name change (#3746) * Revert compile_cmd kwarg name change * Fix binutil tests --- python/tvm/contrib/cc.py | 6 +++--- tests/python/contrib/test_binutil.py | 2 +- 2 files changed, 4 insertions(+), 4 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: 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