From b92e61832f85f35ec36bb0c168bd5e022c1169dc Mon Sep 17 00:00:00 2001
From: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
Date: Tue, 27 Aug 2024 21:58:30 +0530
Subject: [PATCH] build: manually update PyTorch version (#3666)

Set PyTorch and TorchVision version to nightly release 2024-08-25.

Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
---
 projects/pt1/e2e_testing/xfail_sets.py | 8 +++++++-
 pytorch-hash.txt                       | 2 +-
 pytorch-requirements.txt               | 2 +-
 torchvision-requirements.txt           | 2 +-
 4 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/projects/pt1/e2e_testing/xfail_sets.py b/projects/pt1/e2e_testing/xfail_sets.py
index 5c613eae0..ced556cf8 100644
--- a/projects/pt1/e2e_testing/xfail_sets.py
+++ b/projects/pt1/e2e_testing/xfail_sets.py
@@ -505,6 +505,8 @@ FX_IMPORTER_XFAIL_SET = {
     "ViewCollapseDynamicWithAtenSizeIntModule_basic",
     "ViewSizeFromOtherTensor_basic",
     "WeightNormInterfaceModule_basic",
+    "ScaledDotProductAttentionDifferentModule_basic",
+    "ScaledDotProductAttentionSameModule_basic",
 }
 
 FX_IMPORTER_CRASHING_SET = LINALG_CRASHING_SET | {
@@ -2105,7 +2107,6 @@ MAKE_FX_TOSA_PASS_SET = (
         "ViewSizeDimLedAndFollowedByCollapsedOnesModule_basic",
         "ViewSizeDimLedByCollapsedOnesModule_basic",
         "ViewSizeFromOtherTensor_basic",
-        "ScaledDotProductAttentionDifferentModule_basic",
         "RenormModuleFloat32NegativeDim_basic",
         "RenormModuleFloat32_basic",
     }
@@ -2148,6 +2149,11 @@ MAKE_FX_TOSA_PASS_SET = (
     "AdaptiveAvgPool2dFixedKernelStrideSizeStaticModule_basic",
 }
 
+if torch_version_for_comparison() < version.parse("2.5.0.dev"):
+    MAKE_FX_TOSA_PASS_SET = MAKE_FX_TOSA_PASS_SET | {
+        "ScaledDotProductAttentionDifferentModule_basic",
+    }
+
 LTC_CRASHING_SET = {
     # TODO: update test to move all inputs to the lazy device. Otherwise test fails with:
     # Check failed: lazy_tensor Input tensor is not a lazy tensor: CPUBoolType.
diff --git a/pytorch-hash.txt b/pytorch-hash.txt
index 95a62d316..11cae2da8 100644
--- a/pytorch-hash.txt
+++ b/pytorch-hash.txt
@@ -1 +1 @@
-748db193d71a1c29471a87c7841da6a5a0a0dbae
+aa1fc68d51488dab6cf353464ea320e2a0db18f8
diff --git a/pytorch-requirements.txt b/pytorch-requirements.txt
index 75b0983e9..cae8c406e 100644
--- a/pytorch-requirements.txt
+++ b/pytorch-requirements.txt
@@ -1,3 +1,3 @@
 -f https://download.pytorch.org/whl/nightly/cpu/torch/
 --pre
-torch==2.5.0.dev20240818
+torch==2.5.0.dev20240825
diff --git a/torchvision-requirements.txt b/torchvision-requirements.txt
index 5b5890871..c0acddf9a 100644
--- a/torchvision-requirements.txt
+++ b/torchvision-requirements.txt
@@ -1,3 +1,3 @@
 -f https://download.pytorch.org/whl/nightly/cpu/torchvision/
 --pre
-torchvision==0.20.0.dev20240818
+torchvision==0.20.0.dev20240825