From 8d9b3e531099cc7478a4abd944db3b8675326a55 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Thu, 11 May 2023 17:07:32 -0700
Subject: [PATCH] Update driver for cuda 12.1

---
 .github/workflows/validate-windows-binaries.yml | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/.github/workflows/validate-windows-binaries.yml b/.github/workflows/validate-windows-binaries.yml
index 6833e55b2..dd6b97415 100644
--- a/.github/workflows/validate-windows-binaries.yml
+++ b/.github/workflows/validate-windows-binaries.yml
@@ -66,4 +66,8 @@ jobs:
         export ENV_NAME="conda-env-${{ github.run_id }}"
         export TARGET_OS="windows"
         source /c/Jenkins/Miniconda3/etc/profile.d/conda.sh
+        if [[ ${MATRIX_GPU_ARCH_VERSION} == "12.1" ]]; then
+          ./windows/internal/driver_update.bat
+        fi
+
         source ./.github/scripts/validate_binaries.sh