Skip to content
Snippets Groups Projects
Commit 7254a73c authored by Yifan Zhao's avatar Yifan Zhao
Browse files

hpvm-clang moves generated buffer to target directory

parent c1cb4ef9
No related branches found
No related tags found
No related merge requests found
......@@ -98,12 +98,16 @@ def compile_hpvm_c(
_run_opt(hpvm_ll_file, llvm_ll_file, passes, pass_flags),
link_hpvm_rt(link_bitcode_ + [llvm_ll_file], hpvm_rt_linked_file),
]
if tensor_target != "nvdla":
if tensor_target == "nvdla":
# This doesn't generate a bitcode at all.
# Instead it generates hpvm-mod.nvdla and output.protobuf in *current* directory.
# We move them to the working directory.
commands.append(["mv", "hpvm-mod.nvdla", working_dir.as_posix()])
commands.append(["mv", "output.protobuf", working_dir.as_posix()])
else:
commands.append(
link_binary(hpvm_rt_linked_file, output_file, link_dirs, link_libs)
)
for command in commands:
if verbose:
print(" ".join(command))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment