From c2e91452c3bfc75c378fe3efb79c3b8592e7a309 Mon Sep 17 00:00:00 2001
From: pvijayakrish <pvijayakrish@nvidia.com>
Date: Wed, 15 Jan 2025 07:32:27 -0800
Subject: [PATCH] Update DALI args to test on branch 25.01

---
 build.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/build.py b/build.py
index 3b28167ada..6459e16d6b 100755
--- 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):