Skip to content

Commit d6b7b62

Browse files
weberlotqchen
authored andcommitted
Revert compile_cmd kwarg name change (#3746)
* Revert compile_cmd kwarg name change * Fix binutil tests
1 parent 2b975f4 commit d6b7b62

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

python/tvm/contrib/cc.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727
def create_shared(output,
2828
objects,
2929
options=None,
30-
compile_cmd="g++"):
30+
cc="g++"):
3131
"""Create shared library.
3232
3333
Parameters
@@ -41,11 +41,11 @@ def create_shared(output,
4141
options : List[str]
4242
The list of additional options string.
4343
44-
compile_cmd : Optional[str]
44+
cc : Optional[str]
4545
The compiler command.
4646
"""
4747
if sys.platform == "darwin" or sys.platform.startswith("linux"):
48-
_linux_compile(output, objects, options, compile_cmd)
48+
_linux_compile(output, objects, options, cc)
4949
elif sys.platform == "win32":
5050
_windows_shared(output, objects, options)
5151
else:

tests/python/contrib/test_binutil.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ def make_binary():
4343
with open(tmp_source, "w") as f:
4444
f.write(prog)
4545
cc.create_shared(tmp_obj, tmp_source, [],
46-
compile_cmd="{}gcc".format(TOOLCHAIN_PREFIX))
46+
cc="{}gcc".format(TOOLCHAIN_PREFIX))
4747
prog_bin = bytearray(open(tmp_obj, "rb").read())
4848
return prog_bin
4949

0 commit comments

Comments
 (0)