Skip to content

Commit

Permalink
Fix tensorrtllm-build
Browse files Browse the repository at this point in the history
  • Loading branch information
mc-nv committed May 15, 2024
1 parent b9cc3f0 commit 1293f83
Showing 1 changed file with 0 additions and 15 deletions.
15 changes: 0 additions & 15 deletions build.py
Original file line number Diff line number Diff line change
Expand Up @@ -819,21 +819,6 @@ 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"
)
cargs = [
cmake_backend_arg(
"tensorrtllm",
"TRT_LIB_DIR",
None,
"${TRT_ROOT}/targets/${ARCH}-linux-gnu/lib",
),
cmake_backend_arg(
"tensorrtllm", "TRT_INCLUDE_DIR", None, "${TRT_ROOT}/include"
),
]
cargs.append(cmake_backend_enable("tensorrtllm", "USE_CXX11_ABI", True))
return cargs

Expand Down

0 comments on commit 1293f83

Please sign in to comment.