diff --git a/.github/workflows/buildAndTest.yml b/.github/workflows/buildAndTest.yml index 113c9e8e71163..41ce7b414e354 100644 --- a/.github/workflows/buildAndTest.yml +++ b/.github/workflows/buildAndTest.yml @@ -166,6 +166,12 @@ jobs: export PYTHONPATH="$GITHUB_WORKSPACE/build/tools/torch-mlir/python_packages/torch_mlir" python -m e2e_testing.torchscript.main --config=eager_mode -v + - name: Run mhlo e2e integration tests + if: ${{ matrix.os-arch == 'ubuntu-x86_64' && matrix.llvm-build == 'in-tree' }} + run: | + export PYTHONPATH="$GITHUB_WORKSPACE/build/tools/torch-mlir/python_packages/torch_mlir" + python -m e2e_testing.torchscript.main --config=mhlo -v + - name: Run tosa e2e integration tests if: ${{ matrix.os-arch == 'ubuntu-x86_64' && matrix.llvm-build == 'in-tree' }} run: | diff --git a/e2e_testing/torchscript/xfail_sets.py b/e2e_testing/torchscript/xfail_sets.py index c20d137903e06..44d578d08ea23 100644 --- a/e2e_testing/torchscript/xfail_sets.py +++ b/e2e_testing/torchscript/xfail_sets.py @@ -22,7 +22,27 @@ } MHLO_PASS_SET = { + "AdaptiveAvgPool2dNonUnitOutputSizeStaticModule_basic", + "AllBoolFalseModule_basic", + "AllBoolTrueModule_basic", + "AnyBoolFalseModule_basic", + "AnyBoolTrueModule_basic", + "BoolFloatConstantModule_basic", + "BoolIntConstantModule_basic", + "ContainsIntList_False", + "ContainsIntList_True", + "IsFloatingPointFloat_True", + "IsFloatingPointInt_False", + "MeanDimAllReduceKeepdimModule_basic", + "MeanDimAllReduceModule_basic", + "NumToTensorFloatModule_basic", + "SqrtIntConstantModule_basic", + "TorchPrimLoopForLikeModule_basic", + "TorchPrimLoopWhileLikeModule_basic", + "AtenToDeviceModule_basic", "AvgPool2dStaticModule_basic", + "Conv2dWithPaddingDilationStrideStaticModule_basic", + "Convolution2DStaticModule_basic", "ElementwiseCloneContiguousModule_basic", "ElementwiseCloneModule_basic", "ElementwiseBinaryStaticShapeModule_basic", @@ -41,6 +61,8 @@ "SliceModule_basic", "SliceNegIdxModule_basic", "SliceOutOfLowerBoundStartIndexModule_basic", + "SliceOutOfUpperBoundIndexModule_basic", + "SliceStartEqEndModule_basic", "SliceSizeTwoStepModule_basic", "SliceWholeTensorModule_basic", "ReturnTwoTensorF32I64_basic", @@ -128,6 +150,7 @@ "NumpyTRankNDynamicModule_basic", "TModuleRank2_basic", "TensorLiteralModule_basic", + "TensorsConcatModule_basic", "TensorOpaqueLiteralModule_basic", "TransposeIntModule_basic", "TransposeIntNegDimsModule_basic",