From baaab1d49f40eae21bbf4f97da8ab108d47df0b9 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Mon, 6 May 2024 09:25:14 -0700
Subject: [PATCH] Fix cuda windows validations

---
 .github/workflows/validate-windows-binaries.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/validate-windows-binaries.yml b/.github/workflows/validate-windows-binaries.yml
index fed7b1939..cef51b77f 100644
--- a/.github/workflows/validate-windows-binaries.yml
+++ b/.github/workflows/validate-windows-binaries.yml
@@ -127,7 +127,7 @@ jobs:
 
         printf '%s\n' ${{ toJson(inputs.release-matrix) }} > release_matrix.json
         source /c/Jenkins/Miniconda3/etc/profile.d/conda.sh
-        if [[ ${MATRIX_GPU_ARCH_VERSION} == "12.1" ]]; then
+        if [[ ${MATRIX_GPU_ARCH_TYPE} == "cuda" ]]; then
           ./windows/internal/driver_update.bat
         fi
         source ./.github/scripts/validate_binaries.sh