[Task] : Build with 2.8 stable torch.

Author: sahas3

build_tools/ci/install_python_deps.sh

@@ -15,7 +15,7 @@ case $torch_version in
     ;;
   stable)
     echo "::group::installing stable torch"
-    python3 -m pip install --no-cache-dir torch==2.7.1 torchvision --index-url https://download.pytorch.org/whl/cpu
+    python3 -m pip install --no-cache-dir torch==2.8.0 torchvision --index-url https://download.pytorch.org/whl/cpu
     python3 -m pip install --no-cache-dir -r $repo_root/build-requirements.txt
     echo "::endgroup::"
     ;;

lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp

@@ -5193,7 +5193,7 @@ class DecomposeAtenUpsampleNearestVecOp
   using OpRewritePattern<UpsampleVecOp>::OpRewritePattern;
   LogicalResult matchAndRewrite(UpsampleVecOp op,
                                 PatternRewriter &rewriter) const override {
-    Value scales = op.getScaleFactors();
+    // Value scales = op.getScaleFactors();
     static_assert(std::is_same_v<UpsampleVecOp, AtenUpsampleNearest1dVecOp> ||
                   std::is_same_v<UpsampleVecOp, AtenUpsampleNearest2dVecOp>);
     Value cstMode = rewriter.create<Torch::ConstantStrOp>(

projects/pt1/e2e_testing/xfail_sets.py

@@ -487,21 +487,6 @@
     "ViewSizeFromOtherTensor_basic",
     "ViewDtypeStaticModule_basic",
     "WeightNormInterfaceModule_basic",
-    # <<<<<<< HEAD
-    #     # Error: `aten.as_strided` op is not supported
-    #     "ChunkListUnpackDynamic_Module_basic",
-    #     "ChunkListUnpackUnevenDynamic_Module_basic",
-    #     "ChunkListUnpackUneven_Module_basic",
-    #     "ChunkListUnpack_Module_basic",
-    #     "SplitTensorGetItem_Module_basic",
-    #     "SplitTensorLastSmallerModule_basic",
-    #     "SplitTensorListUnpackModule_basic",
-    #     "SplitTensorNegativeDimModule_basic",
-    #     "SplitWithSizesListUnpackModule_basic",
-    #     "SplitWithSizes_Module_basic",
-    #     "AsStridedWithOffsetModule_basic",
-    # =======
-    # >>>>>>> main
     "AdaptiveAvgPool1dGeneralDynamic_basic",
     "AdaptiveAvgPool1dStaticEvenMultiple_basic",
     "AdaptiveAvgPool1dStaticLargerOutput_basic",
@@ -4026,6 +4011,7 @@
     "ReplicationPad1dModule_3DInput_basic",
     "ReplicationPad3dModule_basic",
     "ReplicationPad3dModuleSingleIntPad_basic",
+    # Error: `aten.as_strided` op is not supported
     "AtenAsStridedModule_basic",
     "AtenAsStridedNoStorageOffsetModule_basic",
     "AtenAsStridedUnknownSizeModule_basic",
@@ -4034,6 +4020,7 @@
     "ChunkListUnpackUneven_Module_basic",
     "ChunkListUnpack_Module_basic",
     "NativeGroupNormModule_basic",
+    "AsStridedWithOffsetModule_basic",
     "SplitTensorGetItem_Module_basic",
     "SplitTensorLastSmallerModule_basic",
     "SplitTensorListUnpackModule_basic",
@@ -5268,13 +5255,7 @@
     "AdaptiveMaxPool1dDynamicNoBatch_basic",
     "AdaptiveMaxPool1dDynamic_basic",
     "AdaptiveMaxPool1dStatic_basic",
-    "AsStridedWithOffsetModule_basic",
     "AvgPool2dCHWModule_basic",
-    "ChunkListUnpackDynamic_Module_basic",
-    "ChunkListUnpackUnevenDynamic_Module_basic",
-    "ChunkListUnpackUneven_Module_basic",
-    "ChunkListUnpack_Module_basic",
-    "NativeGroupNormModule_basic",
     "ReduceAnyDimFloatModule_basic",
     "ReflectionPad3dModuleBack_basic",
     "ReflectionPad3dModuleBottom_basic",
@@ -5283,10 +5264,4 @@
     "ReflectionPad3dModuleRight_basic",
     "ReflectionPad3dModuleTop_basic",
     "ReflectionPad3dModule_basic",
-    "SplitTensorGetItem_Module_basic",
-    "SplitTensorLastSmallerModule_basic",
-    "SplitTensorListUnpackModule_basic",
-    "SplitTensorNegativeDimModule_basic",
-    "SplitWithSizesListUnpackModule_basic",
-    "SplitWithSizes_Module_basic",
 }

projects/pt1/python/torch_mlir_e2e_test/configs/fx_importer_backend.py

@@ -32,6 +32,7 @@
         "aten.adaptive_max_pool1d",
         "aten.adaptive_max_pool2d",
         "aten.unflatten.int",
+        "aten.as_strided",
     ],
 }
 

python/torch_mlir/fx_mw.py

@@ -100,6 +100,7 @@ def lower_module(
                 "aten.adaptive_max_pool1d",
                 "aten.adaptive_max_pool2d",
                 "aten.unflatten.int",
+                "aten.as_strided",
             ]
         case "raw":
             output_type = OutputType.RAW