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