Skip to content

Commit 0f796b1

Browse files
committed
Add xfail set for tosa_linalg pipeline.
1 parent fa9d385 commit 0f796b1

File tree

3 files changed

+98
-2
lines changed

3 files changed

+98
-2
lines changed

lib/Conversion/TorchToTosaLinalg/TorchToTosaLinalg.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,9 @@ class ConvertTorchToTosaLinalg
5959
scf::SCFDialect, sparse_tensor::SparseTensorDialect,
6060
tosa::TosaDialect, tensor::TensorDialect,
6161
arith::ArithDialect, complex::ComplexDialect>();
62+
63+
target.addIllegalDialect<Torch::TorchDialect>();
64+
6265
target.addLegalOp<TorchConversion::GetNextSeedOp>();
6366
torch::populateTorchToTosaConversionLegalOps(target);
6467

projects/pt1/e2e_testing/main.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,8 @@
5656
FX_IMPORTER_STABLEHLO_CRASHING_SET,
5757
FX_IMPORTER_TOSA_CRASHING_SET,
5858
FX_IMPORTER_TOSA_XFAIL_SET,
59+
FX_IMPORTER_TOSA_LINALG_CRASHING_SET,
60+
FX_IMPORTER_TOSA_LINALG_XFAIL_SET,
5961
ONNX_TOSA_XFAIL_SET,
6062
ONNX_TOSA_CRASHING_SET,
6163
)
@@ -189,8 +191,8 @@ def main():
189191
crashing_set = FX_IMPORTER_TOSA_CRASHING_SET
190192
elif args.config == "fx_importer_tosa_linalg":
191193
config = FxImporterTestConfig(LinalgOnTensorsTosaBackend(), "tosa_linalg")
192-
xfail_set = FX_IMPORTER_XFAIL_SET
193-
crashing_set = FX_IMPORTER_CRASHING_SET
194+
xfail_set = FX_IMPORTER_TOSA_LINALG_XFAIL_SET
195+
crashing_set = FX_IMPORTER_TOSA_LINALG_CRASHING_SET
194196
elif args.config == "torchdynamo":
195197
# TODO: Enanble runtime verification and extend crashing set.
196198
config = TorchDynamoTestConfig(

projects/pt1/e2e_testing/xfail_sets.py

Lines changed: 91 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5039,3 +5039,94 @@
50395039
# torch.onnx.export produces onnx.MaxPool op with incorrect output shape of 2x6x5x3 instead of 2x6x4x3
50405040
"MaxPool2dStaticCeilModeTrueReduceOutputModule_basic",
50415041
}
5042+
5043+
FX_IMPORTER_TOSA_LINALG_CRASHING_SET = FX_IMPORTER_CRASHING_SET | {
5044+
"BmmIntModule_basic",
5045+
"ConvolutionBackwardModule2DStrided_basic",
5046+
"AvgPool1dIntModule_basic",
5047+
"AvgPool1dStaticModule_basic",
5048+
"AvgPool2dIntModule_basic",
5049+
"ElementwiseAtenLogicalNotOpPromoteModule_basic",
5050+
"ArangeZeroElementOutputModule_basic",
5051+
"Conv2dQInt8Module_basic",
5052+
"Conv2dQInt8Module_depthwise",
5053+
"Conv2dQInt8Module_not_depthwise",
5054+
"AtenIntMM_basic",
5055+
"ElementwiseAddScalarInt8Module_basic",
5056+
"ElementwiseCosIntModule_basic",
5057+
"ElementwiseIntTensorLtFloatTensorModule_basic",
5058+
"ElementwiseFloatTensorGtIntTensorModule_basic",
5059+
"ElementwiseReciprocalIntModule_basic",
5060+
"ElementwiseLog2IntModule_basic",
5061+
"ElementwiseSinIntModule_basic",
5062+
"HstackBasicComplexModule_basic",
5063+
"HstackBasicFloatModule_basic",
5064+
"HstackBasicIntFloatModule_basic",
5065+
"HstackBasicIntModule_basic",
5066+
"ElementwiseRreluEvalModule_basic",
5067+
"ElementwiseRreluEvalStaticModule_basic",
5068+
"ElementwiseRreluTrainModule_basic",
5069+
"ElementwiseRsqrtIntModule_basic",
5070+
"ConvTbcModule_basic",
5071+
"AtenMmIntTypes_basic",
5072+
"ReduceL1NormWithDTypeModule_basic",
5073+
"ReduceSumDimIntListEmptyDimModule_basic",
5074+
"RenormModuleFloat16_basic",
5075+
"MeanDimEmptyDimModule_basic",
5076+
"SliceStaticComplexInputModule_basic",
5077+
"TensorsConcatComplex128IntModule_basic",
5078+
"TensorsConcatComplex64FloatModule_basic",
5079+
"TensorsConcatPromoteDTypeModule_basic",
5080+
"TensorsStackPromoteDTypeModule_basic",
5081+
"SignAndLogarithmOfDeterminantBatchedModule_F32",
5082+
"SignAndLogarithmOfDeterminantDynamicModule_F32",
5083+
"SignAndLogarithmOfDeterminantModule_F32",
5084+
"TrilIndicesNegativeOffsetModule_basic",
5085+
"TriuIndicesModule_basic",
5086+
"NativeBatchNorm1DModule_basic",
5087+
"NativeBatchNorm2DModule_basic",
5088+
"NativeBatchNorm3DModule_basic",
5089+
"NativeBatchNormNoneWeightModule_basic",
5090+
"IndexTensorNegativeIndexModule_basic",
5091+
"UpSampleNearest2dStaticFactor_basic",
5092+
"ElementwiseRreluTrainStaticModule_basic",
5093+
"RsubIntModule_noalpha_basic",
5094+
"TensorsConcatComplex128FloatModule_basic",
5095+
"LinspaceEmptyModule_basic",
5096+
"UpSampleNearest2dStaticSize_basic",
5097+
"UpSampleNearest2dDynamicSize_basic",
5098+
"Permute0RankModule_basic",
5099+
"NumpyTRank0Module_basic",
5100+
"UpSampleNearest2d_basic",
5101+
"InterpolateDynamicModule_sizes_nearest",
5102+
"InterpolateDynamicModule_sizes_bilinear",
5103+
"InterpolateDynamicModule_scales_recompute_bilinear",
5104+
"InterpolateStaticModule_scales_bilinear_align_corners",
5105+
}
5106+
5107+
FX_IMPORTER_TOSA_LINALG_XFAIL_SET = FX_IMPORTER_XFAIL_SET | {
5108+
"Aten_CastLongModule_basic",
5109+
"AvgPool1dFloatModule_basic",
5110+
"AvgPool2dCeilModeTrueModule_basic",
5111+
"AvgPool2dDivisorOverrideModule_basic",
5112+
"AvgPool2dFloatModule_basic",
5113+
"AvgPool2dStaticModule_basic",
5114+
"ConstantBoolParameterModule_basic",
5115+
"CopyWithDifferentDTypesModule_basic",
5116+
"ElementwiseToDtypeF32ToI64Module_basic",
5117+
"LinspaceDtypeModule_basic",
5118+
"MaskedFillTensorFloatValueModule_basic",
5119+
"MobilenetV3Module_basic",
5120+
"PrimsConvertElementTypeModule_basic",
5121+
"RandIntDtypeModule_basic",
5122+
"RandIntLowModule_basic",
5123+
"RandIntModule_basic",
5124+
"RandIntPinMemoryModule_basic",
5125+
"ResNet18Module_basic",
5126+
"ResNet18StaticModule_basic",
5127+
"RollModule_basic",
5128+
"ToCopyWithDTypeFalsePinMemoryModule_basic",
5129+
"ToCopyWithDTypeModule_basic",
5130+
"TrilIndicesAllZerosModule_basic",
5131+
"TriuIndicesAllZerosModule_basic",
5132+
}

0 commit comments

Comments
 (0)