From 3789973cd81d7af00db6b5c43160b06f2b8f7e8f Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjunsuresh1987@gmail.com>
Date: Sun, 18 May 2025 20:15:44 +0530
Subject: [PATCH] Fix for pycuda in nvidia-impl

---
 script/app-mlperf-inference-nvidia/meta.yaml      |  3 +++
 .../meta.yaml                                     | 15 ++++++++++++---
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/script/app-mlperf-inference-nvidia/meta.yaml b/script/app-mlperf-inference-nvidia/meta.yaml
index c1273e286..a5a4e3e53 100644
--- a/script/app-mlperf-inference-nvidia/meta.yaml
+++ b/script/app-mlperf-inference-nvidia/meta.yaml
@@ -365,6 +365,9 @@ variations:
       pycuda:
         version_min: "2024.1"
         version: ""
+      nvidia-inference-server:
+        version: r5.0
+        tags: _mlcommons
   v4.1:
     group: version
     env:
diff --git a/script/build-mlperf-inference-server-nvidia/meta.yaml b/script/build-mlperf-inference-server-nvidia/meta.yaml
index c47c9d819..6ee592a81 100644
--- a/script/build-mlperf-inference-server-nvidia/meta.yaml
+++ b/script/build-mlperf-inference-server-nvidia/meta.yaml
@@ -113,12 +113,9 @@ deps:
   - tags: get,generic-python-lib,_pycuda
     names:
     - pycuda
-    version: "2022.2.2"
     skip_if_env:
       MLC_RUN_STATE_DOCKER:
         - 'yes'
-        - True
-        - 'True'
 
   # Detect opencv-python
   - tags: get,generic-python-lib,_opencv-python
@@ -233,6 +230,9 @@ versions:
   r2.1: {}
   r3.0: {}
   r3.1:
+    add_deps_recursive:
+      pycuda:
+        version: "2022.2.2"
     deps:
       - tags: install,pytorch,from.src,_for-nvidia-mlperf-inference-v3.1
         names:
@@ -247,6 +247,9 @@ versions:
   r4.0:
     default_env:
       BUILD_TRTLLM: 1
+    add_deps_recursive:
+      pycuda:
+        version: "2022.2.2"
     deps:
       - tags: get,generic,sys-util,_nlohmann-json3-dev
       - tags: get,generic,sys-util,_git-lfs
@@ -292,6 +295,9 @@ versions:
   r4.1-dev:
     default_env:
       BUILD_TRTLLM: 1
+    add_deps_recursive:
+      pycuda:
+        version: "2022.2.2"
     deps:
       - tags: get,generic,sys-util,_nlohmann-json3-dev
       - tags: get,generic,sys-util,_git-lfs
@@ -337,6 +343,9 @@ versions:
   r4.1:
     default_env:
       BUILD_TRTLLM: 1
+    add_deps_recursive:
+      pycuda:
+        version: "2022.2.2"
   
   r5.0:
     add_deps_recursive: