Skip to content

Commit

Permalink
Format
Browse files Browse the repository at this point in the history
  • Loading branch information
krishung5 committed Apr 23, 2024
1 parent fb2b2e5 commit f337af3
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion build.py
Original file line number Diff line number Diff line change
Expand Up @@ -820,7 +820,9 @@ def fastertransformer_cmake_args():

def tensorrtllm_cmake_args(images):
cmake_script.cmd("apt-get update && apt-get install -y libcudnn8-dev && ldconfig")
cmake_script.cmd("python3 ../tensorrt_llm/scripts/build_wheel.py --trt_root /usr/local/tensorrt")
cmake_script.cmd(
"python3 ../tensorrt_llm/scripts/build_wheel.py --trt_root /usr/local/tensorrt"
)
cargs = [
cmake_backend_arg(
"tensorrtllm",
Expand Down

0 comments on commit f337af3

Please sign in to comment.