Skip to content

Commit 0cc9d2e

Browse files
committed
Update build.py configuration
1 parent a434122 commit 0cc9d2e

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

build.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -667,6 +667,16 @@ def onnxruntime_cmake_args(images, library_paths):
667667
else TRITON_VERSION_MAP[FLAGS.version][2],
668668
)
669669
]
670+
# If variable is set, ONNX Runtime package will be downloaded from the specified URL
671+
if os.getenv("TRITON_ONNXRUNTIME_PACKAGE_URL"):
672+
cargs.append(
673+
cmake_backend_arg(
674+
"onnxruntime",
675+
"TRITON_ONNXRUNTIME_PACKAGE_URL",
676+
None,
677+
os.getenv("TRITON_ONNXRUNTIME_PACKAGE_URL"),
678+
)
679+
)
670680

671681
# TRITON_ENABLE_GPU is already set for all backends in backend_cmake_args()
672682
# TODO: TPRD-334 TensorRT extension is not currently supported by our manylinux build

0 commit comments

Comments
 (0)