From b3d93ba9a685d0fde69b20bfe4e99833ba0cc7a0 Mon Sep 17 00:00:00 2001
From: Vedaanta Agarwalla <vagarwalla@ipp2-1950.nvidia.com>
Date: Thu, 21 Mar 2024 11:02:44 -0700
Subject: [PATCH] bumps cudnn FE in docker build jobs

---
 .azure/docker-build.yml | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/.azure/docker-build.yml b/.azure/docker-build.yml
index 73233ae78c..7ebc7e1288 100644
--- a/.azure/docker-build.yml
+++ b/.azure/docker-build.yml
@@ -41,13 +41,13 @@ jobs:
       matrix:
         # CUDA 12.1
         'cuda 12.1 | torch 2.2 | cudnn FE v1.1':  # todo: drop updating this image when CI transition to newer FE version
-          {CUDA_VERSION: '12.1.1', TORCH_VERSION: '2.2.1', TRITON_VERSION: '2.2.0', CUDNN_FRONTEND: "1.1.0"}
+          {CUDA_VERSION: '12.1.1', TORCH_VERSION: '2.2.1', TRITON_VERSION: '2.2.0', CUDNN_FRONTEND: "1.2.1"}
         'cuda 12.1 | torch 2.2 | cudnn FE v1.2':
-          {CUDA_VERSION: '12.1.1', TORCH_VERSION: '2.2.1', TRITON_VERSION: '2.2.0', CUDNN_FRONTEND: "1.2.0"}
+          {CUDA_VERSION: '12.1.1', TORCH_VERSION: '2.2.1', TRITON_VERSION: '2.2.0', CUDNN_FRONTEND: "1.2.1"}
         'cuda 12.1 | torch 2.3 /nightly | cudnn FE v1.1':  # todo: drop updating this image when CI transition to newer FE version
-          {CUDA_VERSION: '12.1.1', TORCH_VERSION: 'main', TORCH_INSTALL: 'source', CUDNN_FRONTEND: "1.1.0"}
+          {CUDA_VERSION: '12.1.1', TORCH_VERSION: 'main', TORCH_INSTALL: 'source', CUDNN_FRONTEND: "1.2.1"}
         'cuda 12.1 | torch 2.3 /nightly | cudnn FE v1.2':
-          {CUDA_VERSION: '12.1.1', TORCH_VERSION: 'main', TORCH_INSTALL: 'source', CUDNN_FRONTEND: "1.2.0"}
+          {CUDA_VERSION: '12.1.1', TORCH_VERSION: 'main', TORCH_INSTALL: 'source', CUDNN_FRONTEND: "1.2.1"}
         #'cuda 12.1': # this version - '8.9.5.29-1+cuda12.1' for 'libcudnn8' was not found
     # how much time to give 'run always even if cancelled tasks' before stopping them
     cancelTimeoutInMinutes: "2"