Update DALI args to test on branch 25.01

triton-inference-server · Jan 15, 2025 · c2e9145 · c2e9145
1 parent 27a5fb6
commit c2e9145
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/build.py b/build.py
@@ -813,9 +813,17 @@ def tensorflow_cmake_args(images, library_paths):
 
 
 def dali_cmake_args():
-    return [
+    # return [
+    #     cmake_backend_enable("dali", "TRITON_DALI_SKIP_DOWNLOAD", False),
+    # ]
+    cargs= [
         cmake_backend_enable("dali", "TRITON_DALI_SKIP_DOWNLOAD", False),
+        cmake_backend_arg("dali", "DALI_VERSION", None, "1.45.0"),
+        cmake_backend_arg("dali", "DALI_DOWNLOAD_EXTRA_INDEX_URL", None, "nvidia-dali-cuda120"),
+        cmake_backend_arg("dali", "DALI_DOWNLOAD_PKG_NAME", None, "http://sqrl/nvdl/datasets/dali/pip-dali;"),
+        cmake_backend_arg("dali", "DALI_DOWNLOAD_EXTRA_OPTIONS", None, "--trusted-host sqrl"),
     ]
+    return cargs
 
 
 def fil_cmake_args(images):