We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8d8bd44 commit de3f7bfCopy full SHA for de3f7bf
build.py
@@ -667,6 +667,16 @@ def onnxruntime_cmake_args(images, library_paths):
667
else TRITON_VERSION_MAP[FLAGS.version][2],
668
)
669
]
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
680
681
# TRITON_ENABLE_GPU is already set for all backends in backend_cmake_args()
682
# TODO: TPRD-334 TensorRT extension is not currently supported by our manylinux build
0 commit comments