From f39bea7cf3bbb044c6fb29c1e368d4e3ff29fa9c Mon Sep 17 00:00:00 2001
From: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Date: Fri, 12 Apr 2024 22:19:37 +0200
Subject: [PATCH] Bumps stable torch to `2.2.2` (#165)

---
 .azure/docker-build.yml  | 2 +-
 .azure/gpu-tests.yml     | 4 ++--
 .azure/notebook-runs.yml | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.azure/docker-build.yml b/.azure/docker-build.yml
index 6bb18e928f..a9adefd78b 100644
--- a/.azure/docker-build.yml
+++ b/.azure/docker-build.yml
@@ -41,7 +41,7 @@ jobs:
       matrix:
         # CUDA 12.1
         "cuda 12.1 | torch 2.2 | cudnn FE v1.3":
-          { CUDA_VERSION: "12.1.1", TORCH_VERSION: "2.2.1", TRITON_VERSION: "2.2.0", CUDNN_FRONTEND: "1.3.0" }
+          { CUDA_VERSION: "12.1.1", TORCH_VERSION: "2.2.2", TRITON_VERSION: "2.2.0", CUDNN_FRONTEND: "1.3.0" }
         "cuda 12.1 | torch 2.3 /test | cudnn FE v1.3":
           {
             CUDA_VERSION: "12.1.1",
diff --git a/.azure/gpu-tests.yml b/.azure/gpu-tests.yml
index 69bd9c8621..7b513da730 100644
--- a/.azure/gpu-tests.yml
+++ b/.azure/gpu-tests.yml
@@ -17,10 +17,10 @@ jobs:
       matrix:
         # CUDA 12.1
         "ubuntu22.04 | cuda 12.1 | python 3.10 | torch 2.2 | regular":
-          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.1-apex"
+          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.2-apex"
           CUDA_VERSION_MM: "121"
         "ubuntu22.04 | cuda 12.1 | python 3.10 | torch 2.2 | distributed":
-          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.1-apex"
+          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.2-apex"
           CUDA_VERSION_MM: "121"
           testing: "distributed"
         "ubuntu22.04 | cuda 12.1 | python 3.10 | torch 2.3 | regular":
diff --git a/.azure/notebook-runs.yml b/.azure/notebook-runs.yml
index a82f0673e7..215765d6f5 100644
--- a/.azure/notebook-runs.yml
+++ b/.azure/notebook-runs.yml
@@ -16,7 +16,7 @@ jobs:
     strategy:
       matrix:
         "ubuntu22.04 | cuda 12.1 | torch 2.2":
-          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.1.0-py3.10-pt_2.2.1-apex"
+          docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_2.2.2-apex"
           CUDA_VERSION_MM: "121"
         "ubuntu22.04 | cuda 12.1 | torch-nightly":
           docker-image: "ubuntu22.04-cuda12.1.1-cudnn-fe1.3.0-py3.10-pt_main-apex"