File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -77,12 +77,12 @@ def BUILD_CONFIGS = [
7777 (WHEEL_ARCHS ): " 80-real;86-real;89-real;90-real;100-real;120-real" ,
7878 ],
7979 (CONFIG_LINUX_AARCH64 ): [
80- (WHEEL_EXTRA_ARGS ) : " --extra-cmake-vars WARNING_IS_ERROR=ON" ,
80+ (WHEEL_EXTRA_ARGS ) : " --extra-cmake-vars WARNING_IS_ERROR=ON --extra-cmake-vars NIXL_ROOT=/opt/nvidia/nvda_nixl " ,
8181 (TARNAME ) : " TensorRT-LLM-GH200.tar.gz" ,
8282 (WHEEL_ARCHS ): " 90-real;100-real;120-real" ,
8383 ],
8484 (CONFIG_LINUX_AARCH64_PYBIND ): [
85- (WHEEL_EXTRA_ARGS ) : " --binding_type pybind --extra-cmake-vars WARNING_IS_ERROR=ON" ,
85+ (WHEEL_EXTRA_ARGS ) : " --binding_type pybind --extra-cmake-vars WARNING_IS_ERROR=ON --extra-cmake-vars NIXL_ROOT=/opt/nvidia/nvda_nixl " ,
8686 (TARNAME ) : " pybind-TensorRT-LLM-GH200.tar.gz" ,
8787 (WHEEL_ARCHS ): " 90-real;100-real;120-real" ,
8888 ],
You can’t perform that action at this time.
0 commit comments