From 14e02635b34850f28fbcde9062d82dd5e8ecd430 Mon Sep 17 00:00:00 2001
From: Lujie Duan <lujieduan@google.com>
Date: Tue, 5 Nov 2024 21:45:58 +0000
Subject: [PATCH] Added comments with todo to remove fix

---
 .../third_party_apps_test/applications/dcgm/centos_rhel/install  | 1 +
 .../applications/dcgm/debian_ubuntu/install                      | 1 +
 .../applications/dcgmv1/centos_rhel/install                      | 1 +
 .../applications/dcgmv1/debian_ubuntu/install                    | 1 +
 .../third_party_apps_test/applications/nvml/centos_rhel/install  | 1 +
 .../applications/nvml/debian_ubuntu/install                      | 1 +
 6 files changed, 6 insertions(+)

diff --git a/integration_test/third_party_apps_test/applications/dcgm/centos_rhel/install b/integration_test/third_party_apps_test/applications/dcgm/centos_rhel/install
index b6c65a18e2..9831b116f8 100644
--- a/integration_test/third_party_apps_test/applications/dcgm/centos_rhel/install
+++ b/integration_test/third_party_apps_test/applications/dcgm/centos_rhel/install
@@ -51,6 +51,7 @@ setup_repo() {
 install_cuda_from_package_manager() {
     setup_repo
     install_driver_package
+    # TODO(b/377558109): remove the temporary fix once the repo is updated
     sudo yum -y install cuda-toolkit cuda-demo* 
     verify_driver
 }
diff --git a/integration_test/third_party_apps_test/applications/dcgm/debian_ubuntu/install b/integration_test/third_party_apps_test/applications/dcgm/debian_ubuntu/install
index 170ab4b480..563ef22a13 100644
--- a/integration_test/third_party_apps_test/applications/dcgm/debian_ubuntu/install
+++ b/integration_test/third_party_apps_test/applications/dcgm/debian_ubuntu/install
@@ -34,6 +34,7 @@ case $DEVICE_CODE in
         ;;
     *)
         # For newer GPUs, install the latest version
+        # TODO(b/377558109): remove the temporary fix once the repo is updated
         sudo apt -y install cuda-12-6
         ;;
 esac
diff --git a/integration_test/third_party_apps_test/applications/dcgmv1/centos_rhel/install b/integration_test/third_party_apps_test/applications/dcgmv1/centos_rhel/install
index d1c1263f25..674ef516da 100644
--- a/integration_test/third_party_apps_test/applications/dcgmv1/centos_rhel/install
+++ b/integration_test/third_party_apps_test/applications/dcgmv1/centos_rhel/install
@@ -51,6 +51,7 @@ setup_repo() {
 install_cuda_from_package_manager() {
     setup_repo
     install_driver_package
+    # TODO(b/377558109): remove the temporary fix once the repo is updated
     sudo yum -y install cuda-toolkit cuda-demo*
     verify_driver
 }
diff --git a/integration_test/third_party_apps_test/applications/dcgmv1/debian_ubuntu/install b/integration_test/third_party_apps_test/applications/dcgmv1/debian_ubuntu/install
index 170ab4b480..563ef22a13 100644
--- a/integration_test/third_party_apps_test/applications/dcgmv1/debian_ubuntu/install
+++ b/integration_test/third_party_apps_test/applications/dcgmv1/debian_ubuntu/install
@@ -34,6 +34,7 @@ case $DEVICE_CODE in
         ;;
     *)
         # For newer GPUs, install the latest version
+        # TODO(b/377558109): remove the temporary fix once the repo is updated
         sudo apt -y install cuda-12-6
         ;;
 esac
diff --git a/integration_test/third_party_apps_test/applications/nvml/centos_rhel/install b/integration_test/third_party_apps_test/applications/nvml/centos_rhel/install
index 91352ff9b3..bc30b4c599 100644
--- a/integration_test/third_party_apps_test/applications/nvml/centos_rhel/install
+++ b/integration_test/third_party_apps_test/applications/nvml/centos_rhel/install
@@ -51,6 +51,7 @@ setup_repo() {
 install_cuda_from_package_manager() {
     setup_repo
     install_driver_package
+    # TODO(b/377558109): remove the temporary fix once the repo is updated
     sudo yum -y install cuda-toolkit cuda-demo*
     verify_driver
 }
diff --git a/integration_test/third_party_apps_test/applications/nvml/debian_ubuntu/install b/integration_test/third_party_apps_test/applications/nvml/debian_ubuntu/install
index 5c3cdf85f6..a4dfd08bd5 100644
--- a/integration_test/third_party_apps_test/applications/nvml/debian_ubuntu/install
+++ b/integration_test/third_party_apps_test/applications/nvml/debian_ubuntu/install
@@ -34,6 +34,7 @@ case $DEVICE_CODE in
         ;;
     *)
         # For newer GPUs, install the latest version
+        # TODO(b/377558109): remove the temporary fix once the repo is updated 
         sudo apt -y install cuda-12-6
         ;;
 esac