Merge branch 'feature/backport_ea1_ops' into bump_to_aa7e77ee

.github/workflows/ci.yml

@@ -19,7 +19,7 @@ jobs:
   build-test-linux:
     strategy:
       # AMD: Disable fail-fast to see whether failures are different between stable & nightly
-      # fail-fast: true
+      fail-fast: false
       matrix:
         torch-version: [nightly, stable]
     name: Build and Test (Linux, torch-${{ matrix.torch-version }}, assertions)

projects/pt1/e2e_testing/xfail_sets.py

@@ -36,13 +36,6 @@
     "PowIntIntModule_basic",
 }
 
-if torch_version_for_comparison() < version.parse("2.5.0.dev"):
-    # AttributeError: '_OpNamespace' 'aten' object has no attribute '_safe_softmax'
-    LINALG_XFAIL_SET = LINALG_XFAIL_SET | {
-        "SafeSoftmaxModule_basic",
-        "SafeSoftmaxNonNoneDtypeModule_basic",
-    }
-
 if torch_version_for_comparison() < version.parse("2.5.0.dev"):
     LINALG_XFAIL_SET = LINALG_XFAIL_SET | {
         # Error: 'torch.aten.scaled_dot_product_attention' op expected 8 operands, but found 7
@@ -2461,6 +2454,7 @@
     | {
         ### Tests additionally passing in make_fx_tosa
         "AdaptiveAvgPool1dStaticLargerOutput_basic",
+        "AdaptiveAvgPool1dStaticEvenMultiple_basic",
         "ScaledDotProductAttentionBoolMaskModule_basic",
         "ScaledDotProductAttentionDifferentDynamicCausalModule_basic",
         "ArgminIntModule_basic",
@@ -2517,14 +2511,14 @@
         "MaxPool1dStaticModule_basic",
         "AdaptiveAvgPool1dNonUnitOutputSizeStaticModule_basic",
         "AdaptiveAvgPool1dUnitOutputSizeStaticModule_basic",
-        "AdaptiveAvgPool1dStaticEvenMultiple_basic",
         "CosineSimilarityModule_basic",
         "NativeGroupNormBackwardModule_basic",
         "ReduceFrobeniusNormKeepDimModule_basic",
         "ReduceFrobeniusNormModule_basic",
         "SliceWholeTensorModule_basic",
         "TensorFloatModule_basic",
         "TensorIntModule_basic",
+        "RepeatInterleaveSelfIntModule_basic",
         "AdaptiveAvgPool1dNonUnitOutputSizeStaticModule_basic",
         "AdaptiveAvgPool1dUnitOutputSizeStaticModule_basic",
         "TorchPrimLoopForLikeTensorArgModule_basic",
@@ -2542,21 +2536,22 @@
         "NormalizeModule_basic",
         "ReduceFrobeniusNormKeepDimModule_basic",
         "ReduceFrobeniusNormModule_basic",
-        "ScaledDotProductAttentionBoolMaskModule_basic",
         "SliceEndSleStartStaticModule_basic",
         "ViewSizeDimFollowedByCollapsedOnesModule_basic",
         "ViewSizeDimFollowedByExpandedOnesModule_basic",
         "ViewSizeDimLedAndFollowedByCollapsedOnesModule_basic",
         "ViewSizeDimLedByCollapsedOnesModule_basic",
         "ViewSizeFromOtherTensor_basic",
-        "RepeatInterleaveSelfIntModule_basic",
         "RenormModuleFloat32NegativeDim_basic",
         "RenormModuleFloat32_basic",
     }
 ) - {
     ### Test failing in make_fx_tosa but not in tosa
     # Dynamic shape, has extra unsupported broadcast ops
     "Matmul_3d",
+    # Unimplemented operator 'aten._index_put_impl_.hacked_twin'
+    "IndexPutImpl1DFloatNonAccumulateModule_basic",
+    "IndexPutImpl1DIntNonAccumulateModule_basic",
     # RuntimeError: The size of tensor a (7) must match the size of tensor b (3) at non-singleton dimension 1
     "Add_Module_basic",
     "Conv2dBiasNoPaddingModule_basic",
@@ -2567,8 +2562,6 @@
     "ElementwisePreluModule_basic",
     "ElementwisePreluStaticModule_basic",
     "ElementwiseLogSigmoidModule_basic",
-    "IndexPutImpl1DFloatNonAccumulateModule_basic",
-    "IndexPutImpl1DIntNonAccumulateModule_basic",
     # It appears that you're trying to get value out of a tracing tensor
     # failed to legalize operation 'torch.aten.rrelu_with_noise'
     "ElementwiseRreluEvalModule_basic",
@@ -2596,15 +2589,11 @@
     MAKE_FX_TOSA_PASS_SET = MAKE_FX_TOSA_PASS_SET | {
         "ScaledDotProductAttentionBoolMaskModule_basic",
         "ScaledDotProductAttentionDifferentModule_basic",
-        "ScaledDotProductAttentionDifferentDynamicCausalModule_basic",
         "ScaledDotProductAttentionMaskModule_basic",
         "ScaledDotProductAttentionSameModule_basic",
     }
 
 if torch_version_for_comparison() > version.parse("2.6.0.dev"):
-    MAKE_FX_TOSA_PASS_SET = MAKE_FX_TOSA_PASS_SET | {
-        "ScaledDotProductAttentionDifferentDynamicCausalModule_basic",
-    }
     MAKE_FX_TOSA_PASS_SET = MAKE_FX_TOSA_PASS_SET - {
         "ChunkListUnpackUneven_Module_basic",
         "ChunkListUnpack_Module_basic",

stable-requirements.txt

@@ -1,3 +1,3 @@
 --index-url https://download.pytorch.org/whl/cpu
-torch==2.3.1+cpu
-torchvision==0.18.1+cpu
+torch==2.5.1+cpu
+torchvision==0.20.1+cpu