2021-09-30 00:03:40 +08:00
|
|
|
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
|
|
# See https://llvm.org/LICENSE.txt for license information.
|
|
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
|
|
# Also available under a BSD-style license. See LICENSE.
|
2021-07-01 05:13:21 +08:00
|
|
|
|
|
|
|
# This file describes the sets of tests expected to fail for each config.
|
|
|
|
# This information is deliberately kept in a side table, rather than
|
|
|
|
# in-situ on the test, as a deliberate layering decision: tests should
|
|
|
|
# have unique keys to identify them and enable side tables of various kinds
|
|
|
|
# (this includes down into lower parts of the stack, where a side table
|
|
|
|
# might be used to keep more elaborate sets of testing configurations).
|
|
|
|
|
2022-04-20 03:35:56 +08:00
|
|
|
from torch_mlir_e2e_test.test_suite import COMMON_TORCH_MLIR_LOWERING_XFAILS
|
|
|
|
|
2022-12-07 21:51:24 +08:00
|
|
|
LINALG_XFAIL_SET = COMMON_TORCH_MLIR_LOWERING_XFAILS
|
2021-08-11 07:10:31 +08:00
|
|
|
|
2022-11-18 20:21:19 +08:00
|
|
|
TORCHDYNAMO_XFAIL_SET = {
|
|
|
|
#### General TorchDynamo/PyTorch errors
|
|
|
|
|
|
|
|
# RecursionError: maximum recursion depth exceeded
|
|
|
|
# RuntimeError: Failed running call_function aten.lift_fresh_copy(...
|
2022-11-24 22:36:13 +08:00
|
|
|
# https://github.com/pytorch/pytorch/issues/89627
|
2022-11-18 20:21:19 +08:00
|
|
|
"LiftFreshCopyModule_basic",
|
|
|
|
# TypeError: new_empty(): argument 'size' (position 1) must be tuple of ints, but found element of type NoneType at pos 0
|
|
|
|
# RuntimeError: Failed running call_function aten.convolution_backward(...
|
2022-11-24 22:36:13 +08:00
|
|
|
# https://github.com/pytorch/pytorch/issues/89629
|
2022-11-18 20:21:19 +08:00
|
|
|
"ConvolutionBackwardModule2DPadded_basic",
|
|
|
|
"ConvolutionBackwardModule2D_basic",
|
|
|
|
# RuntimeError: Index tensor must have the same number of dimensions as self tensor
|
|
|
|
# RuntimeError: Failed running call_function aten.nll_loss_backward(...
|
2022-11-24 22:36:13 +08:00
|
|
|
# https://github.com/pytorch/pytorch/issues/89630
|
2022-11-18 20:21:19 +08:00
|
|
|
"NllLossModuleBackward1DMeanWeight_basic",
|
|
|
|
"NllLossModuleBackward1DMean_basic",
|
|
|
|
"NllLossModuleBackward1DSumWeight_basic",
|
|
|
|
"NllLossModuleBackward1DSum_basic",
|
|
|
|
"NllLossModuleBackward1DWeight_basic",
|
|
|
|
"NllLossModuleBackward1D_basic",
|
2022-11-29 22:01:42 +08:00
|
|
|
# TypeError: uniform() missing 2 required keyword-only arguments: 'dtype' and 'device'
|
|
|
|
# RuntimeError: Failed running call_function aten.uniform(...
|
|
|
|
# https://github.com/pytorch/torchdynamo/issues/1954
|
|
|
|
"UniformNoCorrelationModule_basic",
|
2022-11-18 20:21:19 +08:00
|
|
|
# TypeError: expected np.ndarray (got float)
|
2022-11-24 22:36:13 +08:00
|
|
|
# TODO: This is due to returning a scalar float as output from the test.
|
|
|
|
# We should probably just standardize all tests to return tensors.
|
2022-11-18 20:21:19 +08:00
|
|
|
"DivIntModule_basic",
|
|
|
|
|
|
|
|
#### Torch-MLIR internal compiler errors
|
|
|
|
|
|
|
|
# These are probably due to slightly different ops being recorded by
|
|
|
|
# torchdynamo vs. torchscript.
|
|
|
|
|
2022-11-29 22:01:42 +08:00
|
|
|
# No upstream decompositions.
|
2022-11-25 20:33:34 +08:00
|
|
|
# %6:4 = torch.operator "aten._embedding_bag_forward_only"(%1, %3, %5, %false, %int0, %false, %none, %false, %int-1) : (!torch.tensor<*,f32>, !torch.tensor<*,si64>, !torch.tensor<*,si64>, !torch.bool, !torch.int, !torch.bool, !torch.none, !torch.bool, !torch.int) -> (!torch.tensor, !torch.tensor, !torch.tensor, !torch.tensor)
|
2022-11-29 22:01:42 +08:00
|
|
|
# See also: https://github.com/pytorch/torchdynamo/issues/327
|
2022-11-18 20:21:19 +08:00
|
|
|
"AtenEmbeddingBagSumExample_basic",
|
2022-11-25 20:33:34 +08:00
|
|
|
# %1 = torch.operator "aten.scalar_tensor"(%float8.000000e00, %int6, %int0, %cpu, %none) : (!torch.float, !torch.int, !torch.int, !torch.Device, !torch.none) -> !torch.tensor
|
2022-11-18 20:21:19 +08:00
|
|
|
"ElementwiseWhereScalarModule_basic",
|
|
|
|
"ElementwiseWhereScalarOtherModule_basic",
|
|
|
|
"ElementwiseWhereScalarSelfModule_basic",
|
2022-11-25 20:33:34 +08:00
|
|
|
# %7 = torch.operator "aten._index_put_impl_.hacked_twin"(%1, %6, %5, %true, %false) : (!torch.tensor<*,f32>, !torch.list<tensor>, !torch.tensor<*,f32>, !torch.bool, !torch.bool) -> !torch.tensor
|
2022-11-18 20:21:19 +08:00
|
|
|
"IndexPutImpl1DFloatAccumulateModule_basic",
|
|
|
|
"IndexPutImpl1DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPutImpl1DIntAccumulateModule_basic",
|
|
|
|
"IndexPutImpl1DIntNonAccumulateModule_basic",
|
|
|
|
"IndexPutImpl2DFloatAccumulateModule_basic",
|
|
|
|
"IndexPutImpl2DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPutImpl3DFloatAccumulateModule_basic",
|
|
|
|
"IndexPutImpl3DFloatNonAccumulateModule_basic",
|
2022-11-25 20:33:34 +08:00
|
|
|
# %4 = torch.operator "aten.squeeze_.dim"(%3, %int0) : (!torch.tensor<*,f32>, !torch.int) -> !torch.tensor
|
2022-11-18 20:21:19 +08:00
|
|
|
"Matmul_vecmat",
|
2022-11-29 22:01:42 +08:00
|
|
|
|
2022-11-18 20:21:19 +08:00
|
|
|
# https://github.com/llvm/torch-mlir/issues/1611
|
|
|
|
# error: 'tensor.cast' op operand type 'tensor<0xi64>' and result type 'tensor<18xi64>' are cast incompatible
|
|
|
|
"Aten_EmbeddingBagExample_basic",
|
|
|
|
# error: failed to legalize operation 'torch.valsem.aten.bernoulli.float' that was explicitly marked illegal
|
|
|
|
"BernoulliFloatModule_basic",
|
|
|
|
# error: failed to legalize operation 'torch.aten.view' that was explicitly marked illegal
|
|
|
|
"ElementwiseFlattenBroadcastModule_basic",
|
|
|
|
"FlattenRank0Module_basic",
|
2022-11-29 22:01:42 +08:00
|
|
|
"UniformModule_basic",
|
2022-11-18 20:21:19 +08:00
|
|
|
# error: failed to materialize conversion for result #0 of operation 'torch.aten.t' that remained live after conversion
|
|
|
|
"TModuleRank1_basic",
|
|
|
|
}
|
|
|
|
|
2022-08-23 16:47:21 +08:00
|
|
|
MHLO_PASS_SET = {
|
2022-12-22 17:12:52 +08:00
|
|
|
"AdaptiveAvgPool2dNonUnitOutputSizeStaticModule_basic",
|
2022-09-20 22:31:24 +08:00
|
|
|
"ArangeDtypeFloatModule_basic",
|
|
|
|
"ArangeDtypeIntModule_basic",
|
|
|
|
"ArangeFalsePinMemoryModule_basic",
|
|
|
|
"ArangeFloatModule_basic",
|
|
|
|
"ArangeIntModule_basic",
|
|
|
|
"ArangeNegativeStartFloatModule_basic",
|
|
|
|
"ArangeNegativeStartIntModule_basic",
|
|
|
|
"ArangeStartFloatModule_basic",
|
|
|
|
"ArangeStartIntModule_basic",
|
|
|
|
"ArangeStartNegativeStepFloatModule_basic",
|
|
|
|
"ArangeStartNegativeStepIntModule_basic",
|
|
|
|
"ArangeStartStepFloatModule_basic",
|
|
|
|
"ArangeStartStepIntModule_basic",
|
|
|
|
"ArangeZeroElementOutputModule_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"BmmModule_basic",
|
|
|
|
"BroadcastToModule_basic",
|
|
|
|
"BroadcastToSameRankStaticModule_basic",
|
|
|
|
"BroadcastZeroRankInputStaticModule_basic",
|
2022-11-24 14:28:34 +08:00
|
|
|
"ElementwiseAtenWhereSelfModule_basic",
|
2022-09-16 15:09:21 +08:00
|
|
|
"ElementwiseClampModule_basic",
|
|
|
|
"ElementwiseClampMinModule_basic",
|
|
|
|
"ElementwiseClampMaxModule_basic",
|
2022-09-08 10:15:36 +08:00
|
|
|
"ElementwiseExpModule_basic",
|
|
|
|
"ElementwiseLogModule_basic",
|
|
|
|
"ElementwiseNegModule_basic",
|
2022-12-22 10:13:59 +08:00
|
|
|
"ElementwiseRsqrtModule_basic",
|
|
|
|
"ElementwiseSigmoidModule_basic",
|
2022-09-08 10:15:36 +08:00
|
|
|
"ElementwiseSqrtModule_basic",
|
|
|
|
"ElementwiseUnaryModule_basic",
|
|
|
|
"ElementwiseUnsqueezeNegDimsModule_basic",
|
2022-09-23 10:24:36 +08:00
|
|
|
"ElementwiseToDtypeF32ToI64Module_basic",
|
2022-09-23 20:39:15 +08:00
|
|
|
"ElementwiseAddModule_basic",
|
|
|
|
"ElementwiseAddScalarFloatModule_basic",
|
|
|
|
"ElementwiseAddScalarInt64Module_basic",
|
|
|
|
"ElementwiseAddScalarIntModule_basic",
|
|
|
|
"ElementwiseDivScalarModule_basic",
|
|
|
|
"ElementwiseEqDiffWidthScalarModule_basic",
|
|
|
|
"ElementwiseEqFloatScalarModule_basic",
|
|
|
|
"ElementwiseEqIntScalarModule_basic",
|
|
|
|
"ElementwiseErfModule_basic",
|
|
|
|
"ElementwiseGeluModule_basic",
|
|
|
|
"ElementwiseGtFloatScalarModule_basic",
|
|
|
|
"ElementwiseGtIntScalarModule_basic",
|
|
|
|
"ElementwiseGtMixed2ScalarModule_basic",
|
2022-11-24 14:28:34 +08:00
|
|
|
"ElementwiseGeFloatIntScalarModule_basic",
|
|
|
|
"ElementwiseGeFloatScalarModule_basic",
|
|
|
|
"ElementwiseGeIntScalarModule_basic",
|
|
|
|
"ElementwiseGeMixedIntScalarModule_basic",
|
|
|
|
"ElementwiseLeFloatIntScalarModule_basic",
|
|
|
|
"ElementwiseLeFloatScalarModule_basic",
|
|
|
|
"ElementwiseLeIntScalarModule_basic",
|
|
|
|
"ElementwiseLeMixedIntScalarModule_basic",
|
2022-09-23 20:39:15 +08:00
|
|
|
"ElementwiseLtDiffWidthScalarModule_basic",
|
|
|
|
"ElementwiseLtFloatScalarModule_basic",
|
|
|
|
"ElementwiseLtIntScalarModule_basic",
|
|
|
|
"ElementwiseMulScalarModule_basic",
|
|
|
|
"ElementwiseMulScalarModule_float",
|
|
|
|
"ElementwiseMulScalarModule_int",
|
|
|
|
"ElementwiseNeFloatTensorModule_basic",
|
|
|
|
"ElementwiseNeIntScalarModule_basic",
|
|
|
|
"ElementwiseReciprocalModule_basic",
|
|
|
|
"ElementwiseRelu6Module_basic",
|
|
|
|
"ElementwiseReluModule_basic",
|
2022-11-24 14:28:34 +08:00
|
|
|
"ElementwiseRemainderScalarModule_Bool_basic",
|
|
|
|
"ElementwiseRemainderScalarModule_Float_basic",
|
|
|
|
"ElementwiseRemainderScalarModule_Int_Float_basic",
|
|
|
|
"ElementwiseRemainderScalarModule_Int_basic",
|
2022-09-23 20:39:15 +08:00
|
|
|
"ElementwiseSubScalarFloatModule_basic",
|
|
|
|
"ElementwiseSubScalarIntModule_basic",
|
2022-11-24 14:28:34 +08:00
|
|
|
"ElementwiseWhereScalarModule_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"EmbeddingModule1DIndices_basic",
|
|
|
|
"EmbeddingModuleI32Static_basic",
|
|
|
|
"EmbeddingModuleI32_basic",
|
|
|
|
"EmbeddingModuleI64_basic",
|
2022-11-30 19:14:48 +08:00
|
|
|
"EmbeddingModuleF16_basic",
|
2022-09-23 10:24:36 +08:00
|
|
|
"ExpandAsIntModule_basic",
|
2022-09-08 10:15:36 +08:00
|
|
|
"ExpandModule_basic",
|
2022-09-23 10:24:36 +08:00
|
|
|
"FullLikeModuleDefaultDtype_basic",
|
|
|
|
"FullLikeModuleFalsePinMemory_basic",
|
|
|
|
"FullLikeModuleFloat2D_basic",
|
|
|
|
"FullLikeModuleFloat3DStatic_basic",
|
|
|
|
"FullLikeModuleFloat3D_basic",
|
|
|
|
"FullLikeModuleInt2DStatic_basic",
|
|
|
|
"FullLikeModuleInt2D_basic",
|
|
|
|
"FullLikeModuleInt3D_basic",
|
|
|
|
"FullModuleDefaultDtype_basic",
|
|
|
|
"FullModuleFalsePinMemory_basic",
|
|
|
|
"FullModuleFloat2D_basic",
|
|
|
|
"FullModuleFloat3D_basic",
|
|
|
|
"FullModuleInt2D_basic",
|
|
|
|
"FullModuleInt3D_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"GatherStaticModule_basic",
|
|
|
|
"GatherModule_basic",
|
|
|
|
"Gather2DInputModdule_basic",
|
|
|
|
"GatherRandomIndexModule_basic",
|
2022-12-21 20:09:43 +08:00
|
|
|
"GeluBackwardModule_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"HardTanhIntModule_basic",
|
|
|
|
"HardTanhModule_basic",
|
|
|
|
"HardsigmoidModule_basic",
|
|
|
|
"HardsigmoidRandomModule_basic",
|
|
|
|
"IndexSelectDynamicIndexSizeModule_basic",
|
|
|
|
"IndexSelectSingleIdxModule_basic",
|
|
|
|
"IndexSelectTwoIdxModule_basic",
|
|
|
|
"IndexSelectWholeDimensionModule_basic",
|
|
|
|
"IndexSelectWholeTensorModule_basic",
|
2022-12-22 10:13:59 +08:00
|
|
|
"LayerNormLastDimModule_basic",
|
|
|
|
"LayerNormModule_basic",
|
|
|
|
"LayerNormNormalizeOverAllDimsModule_basic",
|
2022-09-08 10:15:36 +08:00
|
|
|
"MatmulBroadcastBatchDim_basic",
|
|
|
|
"MatmulSingleDynamicBatchDim_basic",
|
|
|
|
"Matmul_3d",
|
|
|
|
"Matmul_4d",
|
2022-11-23 15:02:41 +08:00
|
|
|
"MeanDimEmptyDimModule_basic",
|
2022-09-08 10:15:36 +08:00
|
|
|
"MeanDtypeModule_basic",
|
2022-11-23 15:02:41 +08:00
|
|
|
"MeanDynamicSizesModule_basic",
|
|
|
|
"MeanLargeInputModule_basic",
|
|
|
|
"MeanModule_basic",
|
2022-09-08 10:15:36 +08:00
|
|
|
"MmTanhModule_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"Mv_basic",
|
2022-12-22 10:13:59 +08:00
|
|
|
"NativeLayerNormModule4D_basic",
|
|
|
|
"NativeLayerNormModule_basic",
|
2022-11-23 15:02:41 +08:00
|
|
|
"PrimsConvertElementTypeModule_basic",
|
2022-09-08 10:15:36 +08:00
|
|
|
"ReduceFrobeniusNormKeepDimModule_basic",
|
2022-11-23 15:02:41 +08:00
|
|
|
"ReduceSumDimIntListElementTypeBoolModule_basic",
|
|
|
|
"ReduceSumElementTypeBoolModule_basic",
|
|
|
|
"ReduceSumDimIntListEmptyDimModule_basic",
|
2022-09-08 10:15:36 +08:00
|
|
|
"ReduceSumDimIntListDtypeFloatModule_basic",
|
|
|
|
"ReduceSumDimIntListDtypeIntModule_basic",
|
|
|
|
"ReduceSumDimIntListKeepDimFloatModule_basic",
|
|
|
|
"ReduceSumDimIntListKeepDimIntModule_basic",
|
|
|
|
"ReduceSumDtypeFloatModule_basic",
|
|
|
|
"ReduceSumDtypeIntModule_basic",
|
|
|
|
"SelectIntModule_basic",
|
|
|
|
"SliceSingleIdxModule_basic",
|
|
|
|
"SqueezeDimModule_dynamic",
|
|
|
|
"SqueezeDimModule_negDim",
|
2022-12-22 10:13:59 +08:00
|
|
|
"ToCopyBoolDTypeStaticModule_basic",
|
|
|
|
"ToCopyModule_basic",
|
|
|
|
"ToCopyWithDTypeFalsePinMemoryModule_basic",
|
|
|
|
"ToCopyWithDTypeModule_basic",
|
2022-09-08 10:15:36 +08:00
|
|
|
"ReduceFrobeniusNormModule_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
"FlattenStaticModule_basic",
|
|
|
|
"FlattenRank0Module_basic",
|
|
|
|
"TensorsConcatNegativeDimModule_basic",
|
2022-09-23 10:24:36 +08:00
|
|
|
"LiftFreshCopyModule_basic",
|
|
|
|
"Mlp2LayerModuleNoBias_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
"NumelModule_basic",
|
|
|
|
"SqueezeModule_allUnitDim",
|
|
|
|
"SqueezeDimModule_unitDim",
|
2022-09-28 02:08:14 +08:00
|
|
|
"ViewCollapseOnesMiddleModule_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"ViewDoubleMergeStaticModule_basic",
|
|
|
|
"ViewExpandDynamicDimModule_basic",
|
|
|
|
"ViewFlattenAndExpandModule_basic",
|
2022-09-28 02:08:14 +08:00
|
|
|
"ViewFiveTestStaticModule_basic",
|
|
|
|
"ViewOffsetTestStaticModule_basic",
|
|
|
|
"ViewTwoFiveThreeStaticModule_basic",
|
|
|
|
"ViewTwoToThreeStaticModule_basic",
|
|
|
|
"ViewExpandOnesMiddleOppModule_basic",
|
|
|
|
"ViewOffsetBackwardTestStaticModule_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
"NumToTensorFloatModule_basic",
|
|
|
|
"AtenToDeviceModule_basic",
|
|
|
|
"AvgPool2dStaticModule_basic",
|
|
|
|
"Conv2dWithPaddingDilationStrideStaticModule_basic",
|
|
|
|
"Convolution2DStaticModule_basic",
|
2022-08-27 06:29:07 +08:00
|
|
|
"ConvolutionModule2DTransposeStridedStatic_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
"ElementwiseCloneContiguousModule_basic",
|
|
|
|
"ElementwiseCloneModule_basic",
|
|
|
|
"ElementwiseBinaryStaticShapeModule_basic",
|
|
|
|
"ReturnThreeTensorFloat32_basic",
|
|
|
|
"BoolTensorReturnFalseModule_basic",
|
|
|
|
"BoolTensorReturnTrueModule_basic",
|
|
|
|
"BoolTensorReturnMixedModule_basic",
|
|
|
|
"SqueezeModule_static",
|
|
|
|
"TModuleRank1_basic",
|
|
|
|
"TModuleRank0_basic",
|
|
|
|
"ElementwiseToDtypeIdentityModule_basic",
|
2022-11-24 21:02:59 +08:00
|
|
|
"View1DFoldModule_basic",
|
|
|
|
"UnsafeView1DFoldModule_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"RsubFloatModule_basic",
|
|
|
|
"RsubFloatModule_noalpha_basic",
|
|
|
|
"RsubIntModule_basic",
|
|
|
|
"RsubIntModule_noalpha_basic",
|
|
|
|
"SliceStaticModule_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
"SliceModule_basic",
|
|
|
|
"SliceNegIdxModule_basic",
|
|
|
|
"SliceOutOfLowerBoundStartIndexModule_basic",
|
|
|
|
"SliceOutOfUpperBoundIndexModule_basic",
|
|
|
|
"SliceStartEqEndModule_basic",
|
|
|
|
"SliceSizeTwoStepModule_basic",
|
|
|
|
"SliceWholeTensorModule_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"SqueezeDimModule_static",
|
|
|
|
"SqueezeDimModule_identity",
|
|
|
|
"SqueezeModule_broadcast",
|
2022-08-23 16:47:21 +08:00
|
|
|
"ReturnTwoTensorF32I64_basic",
|
|
|
|
"Matmul4dStatic_basic",
|
|
|
|
"Matmul_dot",
|
|
|
|
"Matmul_2d",
|
|
|
|
"Matmul_matvec",
|
|
|
|
"Matmul_vecmat",
|
|
|
|
"MaxPool2dWithIndicesStaticModule_basic",
|
|
|
|
"MmDagModule_basic",
|
|
|
|
"MmModule_basic",
|
|
|
|
"MmModule_chained",
|
|
|
|
"MaxPool2dStaticModule_basic",
|
|
|
|
"PermuteModule_basic",
|
|
|
|
"PermuteNegativeIndexModule_basic",
|
2022-09-23 10:24:36 +08:00
|
|
|
"ReduceSumDimIntListKeepDimNegativeDimStaticModule_basic",
|
|
|
|
"ZerosLikeModule_defaultDtype",
|
|
|
|
"ZerosLikeModule_falsePinMemory",
|
|
|
|
"ZerosLikeModule_float",
|
|
|
|
"ZerosLikeModule_int",
|
2022-08-23 16:47:21 +08:00
|
|
|
"ZerosModuleDefaultDtype_basic",
|
|
|
|
"ZerosModuleInt2D_basic",
|
|
|
|
"ZerosModuleInt3D_basic",
|
|
|
|
"ZerosModuleFloat2D_basic",
|
|
|
|
"ZerosModuleFloat3D_basic",
|
|
|
|
"ZerosModuleFalsePinMemory_basic",
|
|
|
|
"OnesModuleDefaultDtype_basic",
|
|
|
|
"OnesModuleInt_basic",
|
|
|
|
"OnesModuleFloat_basic",
|
|
|
|
"OnesModuleFalsePinMemory_basic",
|
2022-09-23 10:24:36 +08:00
|
|
|
"OnesLikeModule_defaultDtype",
|
|
|
|
"OnesLikeModule_falsePinMemory",
|
|
|
|
"OnesLikeModule_float",
|
|
|
|
"OnesLikeModule_int",
|
2022-08-23 16:47:21 +08:00
|
|
|
"NewZerosModuleDefaultDtype_basic",
|
|
|
|
"NewZerosModuleInt2D_basic",
|
|
|
|
"NewZerosModuleInt3D_basic",
|
|
|
|
"NewZerosModuleFloat2D_basic",
|
|
|
|
"NewZerosModuleFloat3D_basic",
|
|
|
|
"NewZerosModuleFalsePinMemory_basic",
|
|
|
|
"NewOnesModuleDefaultDtype_basic",
|
|
|
|
"NewOnesModuleInt2D_basic",
|
|
|
|
"NewOnesModuleInt3D_basic",
|
|
|
|
"NewOnesModuleFloat2D_basic",
|
|
|
|
"NewOnesModuleFloat3D_basic",
|
|
|
|
"NewOnesModuleFalsePinMemory_basic",
|
|
|
|
"DropoutEvalIntModule_basic",
|
|
|
|
"DropoutEvalFloatModule_basic",
|
|
|
|
"ContiguousModule_basic",
|
|
|
|
"DropoutModule_basic",
|
|
|
|
"ViewCollapseModule_basic",
|
2022-11-24 21:02:59 +08:00
|
|
|
"ViewCollapseInferredDimModule_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
"ViewDynamicExpandCollapseModule_basic",
|
|
|
|
"ViewDynamicExpandModule_basic",
|
|
|
|
"ViewExpandModule_basic",
|
|
|
|
"ViewExpandOnesModule_basic",
|
|
|
|
"ViewExpandOnesBeforeAndAfterModule_basic",
|
|
|
|
"ViewExpandOnesMiddleModule_basic",
|
|
|
|
"ViewExpandCollapseModule_basic",
|
|
|
|
"ViewExpandCollapseWithOnesModule_basic",
|
2022-11-24 21:02:59 +08:00
|
|
|
"ViewExpandInferredDimModule_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
"ViewNoChangeStaticModule_basic",
|
|
|
|
"ViewNoChange1dModule_basic",
|
|
|
|
"ViewNoChange2dModule_basic",
|
|
|
|
"ViewNoChange3dModule_basic",
|
|
|
|
"UnsafeViewExpandModule_basic",
|
|
|
|
"ReduceMaxAllDims_basic",
|
|
|
|
"ReduceMaxFloatModule_basic",
|
|
|
|
"ReduceMaxSignedIntModule_basic",
|
|
|
|
"ReduceMaxUnsignedIntModule_basic",
|
|
|
|
"ReduceSumDimIntListFloatModule_basic",
|
|
|
|
"ReduceSumDimIntListIntModule_basic",
|
|
|
|
"ReduceSumFloatModule_basic",
|
|
|
|
"ReduceSumSignedIntModule_basic",
|
|
|
|
"ReduceSumUnsignedIntModule_basic",
|
|
|
|
"RepeatModule_basic",
|
|
|
|
"ReshapeAliasCollapseModule_basic",
|
|
|
|
"ReshapeAliasExpandModule_basic",
|
|
|
|
"ReshapeExpandModule_basic",
|
2022-08-24 08:36:05 +08:00
|
|
|
"RollModule_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
"TestMultipleTensorReturn_basic",
|
|
|
|
"AdaptiveAvgPool2dUnitOutputSizeStaticModule_basic",
|
|
|
|
"BaddbmmStaticModule_basic",
|
|
|
|
"BaddbmmBroadcast1DInputModule_basic",
|
|
|
|
"BaddbmmBroadcast2DInputModule_basic",
|
|
|
|
"NarrowHorizontalTest2_basic",
|
|
|
|
"NarrowHorizontalTest_basic",
|
|
|
|
"NarrowVerticalTest2_basic",
|
|
|
|
"NarrowVerticalTest_basic",
|
|
|
|
"NumToTensorIntModule_basic",
|
|
|
|
"NumpyTRank0Module_basic",
|
|
|
|
"NumpyTRank1Module_basic",
|
|
|
|
"NumpyTRank2Module_basic",
|
|
|
|
"NumpyTRankNStaticModule_basic",
|
|
|
|
"NumpyTRankNDynamicModule_basic",
|
|
|
|
"TModuleRank2_basic",
|
|
|
|
"TensorLiteralModule_basic",
|
|
|
|
"TensorsConcatModule_basic",
|
|
|
|
"TensorOpaqueLiteralModule_basic",
|
|
|
|
"TransposeIntModule_basic",
|
|
|
|
"TransposeIntNegDimsModule_basic",
|
2022-09-23 10:24:36 +08:00
|
|
|
"ToDtypeBoolLayoutNoneModule_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"ToDtypeBoolLayoutNoneStaticModule_basic",
|
2022-09-23 10:24:36 +08:00
|
|
|
"ToDtypeLayoutNoneModule_basic",
|
|
|
|
"ToDtypeLayoutStridedModule_basic",
|
|
|
|
"TypeAsSameModule_basic",
|
|
|
|
"TypeConversionF32ToF64Module_basic",
|
|
|
|
"TypeConversionF64ToF32Module_basic",
|
|
|
|
"TypeConversionI1ToF32Module_basic",
|
|
|
|
"TypeConversionI1ToF64Module_basic",
|
|
|
|
"TypeConversionI1ToI32Module_basic",
|
|
|
|
"TypeConversionI1ToI64Module_basic",
|
|
|
|
"TypeConversionI32ToI64Module_basic",
|
|
|
|
"TypeConversionI64ToI32Module_basic",
|
2022-11-21 21:50:35 +08:00
|
|
|
"TypePromotionAlphaWiderModule_basic",
|
|
|
|
"TypePromotionSameCategoryZeroRankWider_basic",
|
|
|
|
"TypePromotionZeroRankHigherCategoryModule_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
"OnesModuleCPUDevice_basic",
|
|
|
|
"Permute0RankModule_basic",
|
|
|
|
"UnsafeViewCollapseModule_basic",
|
|
|
|
"UnsafeViewDynamicExpandModule_basic",
|
2022-10-11 17:52:01 +08:00
|
|
|
"AtenRoundIntModule_basic",
|
2022-12-01 23:46:17 +08:00
|
|
|
"TestF16Return_basic",
|
2022-08-23 16:47:21 +08:00
|
|
|
}
|
|
|
|
|
2021-10-08 10:07:03 +08:00
|
|
|
# Write the TOSA set as a "passing" set as it is very early in development
|
|
|
|
# and very few tests work yet.
|
2021-10-26 02:43:21 +08:00
|
|
|
TOSA_PASS_SET = {
|
2022-09-21 06:07:46 +08:00
|
|
|
"ElementwiseCloneContiguousModule_basic",
|
|
|
|
"ElementwiseCloneModule_basic",
|
2021-10-26 04:43:54 +08:00
|
|
|
"ElementwiseUnaryModule_basic",
|
2021-12-04 15:31:38 +08:00
|
|
|
"ElementwiseBinaryModule_basic",
|
2021-10-29 01:09:12 +08:00
|
|
|
"ElementwiseSigmoidModule_basic",
|
2022-03-24 04:35:43 +08:00
|
|
|
"ElementwiseExpModule_basic",
|
2022-08-31 03:44:00 +08:00
|
|
|
"ElementwiseReluModule_basic",
|
2021-11-11 11:03:36 +08:00
|
|
|
"ElementwiseFloorModule_basic",
|
|
|
|
"ElementwiseLogModule_basic",
|
2021-12-29 10:11:07 +08:00
|
|
|
"ElementwiseBinaryStaticShapeModule_basic",
|
2022-02-09 04:57:23 +08:00
|
|
|
"ElementwiseMinimumModule_basic",
|
|
|
|
"ElementwiseMinimumIntModule_basic",
|
|
|
|
"ElementwiseMaximumModule_basic",
|
|
|
|
"ElementwiseMaximumIntModule_basic",
|
2022-09-28 02:08:14 +08:00
|
|
|
"ViewDoubleMergeStaticModule_basic",
|
|
|
|
"ViewCollapseOnesMiddleModule_basic",
|
|
|
|
"ViewFiveTestStaticModule_basic",
|
|
|
|
"ViewOffsetTestStaticModule_basic",
|
|
|
|
"ViewTwoFiveThreeStaticModule_basic",
|
|
|
|
"ViewTwoToThreeStaticModule_basic",
|
|
|
|
"ViewExpandOnesMiddleOppModule_basic",
|
|
|
|
"ViewOffsetBackwardTestStaticModule_basic",
|
2021-11-12 08:15:58 +08:00
|
|
|
"TanhBackward_basic",
|
2021-11-24 14:25:59 +08:00
|
|
|
"ElementwiseAddModule_basic",
|
2021-11-08 23:56:40 +08:00
|
|
|
"ReturnThreeTensorFloat32_basic",
|
2021-11-25 06:01:48 +08:00
|
|
|
"AddCMulModule_basic",
|
|
|
|
"AddCDivModule_basic",
|
2021-12-01 05:48:10 +08:00
|
|
|
"SqueezeModule_broadcast",
|
2021-12-08 22:05:02 +08:00
|
|
|
"BoolTensorReturnFalseModule_basic",
|
|
|
|
"BoolTensorReturnTrueModule_basic",
|
|
|
|
"BoolTensorReturnMixedModule_basic",
|
2022-04-14 01:28:27 +08:00
|
|
|
"BoolTensorHandleSignless_basic",
|
2021-12-15 02:03:58 +08:00
|
|
|
"ElementwiseRsqrtModule_basic",
|
2021-12-16 13:40:29 +08:00
|
|
|
"SqueezeModule_static",
|
|
|
|
"SqueezeModule_noUnitDim",
|
|
|
|
"SqueezeModule_allUnitDim",
|
2021-12-17 12:08:07 +08:00
|
|
|
"TModuleRank1_basic",
|
|
|
|
"TModuleRank0_basic",
|
2021-12-23 20:04:29 +08:00
|
|
|
"ElementwiseToDtypeIdentityModule_basic",
|
2022-08-11 07:24:02 +08:00
|
|
|
"AtenToDeviceModule_basic",
|
2022-11-24 21:02:59 +08:00
|
|
|
"View1DFoldModule_basic",
|
|
|
|
"UnsafeView1DFoldModule_basic",
|
2022-01-07 00:31:29 +08:00
|
|
|
"SqueezeDimModule_static",
|
|
|
|
"SqueezeDimModule_identity",
|
|
|
|
"SqueezeDimModule_unitDim",
|
2022-01-10 21:24:08 +08:00
|
|
|
"ReturnTwoTensorF32I64_basic",
|
2022-01-15 03:05:11 +08:00
|
|
|
"ElementwisePowModule_basic",
|
2022-01-20 04:10:04 +08:00
|
|
|
"BmmModule_basic",
|
2022-02-19 01:56:48 +08:00
|
|
|
"MmDagModule_basic",
|
2022-06-16 23:45:10 +08:00
|
|
|
"Matmul4dStatic_basic",
|
2022-01-20 04:10:04 +08:00
|
|
|
"Matmul_dot",
|
|
|
|
"Matmul_3d",
|
2022-06-14 20:31:30 +08:00
|
|
|
"RsubFloatModule_basic",
|
|
|
|
"RsubFloatModule_noalpha_basic",
|
2022-01-21 02:58:30 +08:00
|
|
|
"ElementwiseGtFloatScalarModule_basic",
|
|
|
|
"ElementwiseGtIntScalarModule_basic",
|
|
|
|
"ElementwiseGtMixed2ScalarModule_basic",
|
|
|
|
"ElementwiseGtFloatTensorModule_basic",
|
|
|
|
"ElementwiseGtIntTensorModule_basic",
|
|
|
|
"ElementwiseLtFloatScalarModule_basic",
|
|
|
|
"ElementwiseLtIntScalarModule_basic",
|
|
|
|
"ElementwiseLtDiffWidthScalarModule_basic",
|
|
|
|
"ElementwiseLtFloatTensorModule_basic",
|
|
|
|
"ElementwiseLtIntTensorModule_basic",
|
|
|
|
"ElementwiseEqFloatScalarModule_basic",
|
|
|
|
"ElementwiseEqIntScalarModule_basic",
|
|
|
|
"ElementwiseEqDiffWidthScalarModule_basic",
|
|
|
|
"ElementwiseEqFloatTensorModule_basic",
|
|
|
|
"ElementwiseEqIntTensorModule_basic",
|
|
|
|
"ElementwiseMulScalarModule_int",
|
|
|
|
"ElementwiseMulScalarModule_float",
|
|
|
|
"ElementwiseMulTensorIntModule_basic",
|
|
|
|
"ElementwiseDivScalarModule_basic",
|
|
|
|
"ElementwiseSubScalarFloatModule_basic",
|
|
|
|
"ElementwiseAddScalarFloatModule_basic",
|
|
|
|
"ElementwiseMulScalarModule_float",
|
|
|
|
"ElementwiseCeilModule_basic",
|
|
|
|
"ElementwiseReciprocalModule_basic",
|
2022-09-07 14:58:42 +08:00
|
|
|
"ElementwiseNotIntegerModule_basic",
|
|
|
|
"ElementwiseNotInt32Module_basic",
|
2022-01-21 02:58:30 +08:00
|
|
|
"TypePromotionAlphaWiderModule_basic",
|
2022-01-27 11:16:13 +08:00
|
|
|
"Conv2dWithPaddingDilationStrideStaticModule_basic",
|
2022-02-25 03:41:55 +08:00
|
|
|
"BatchNorm1DModule_basic",
|
2022-07-16 00:35:59 +08:00
|
|
|
"BatchNorm1DWith2DInputModule_basic",
|
2022-02-25 03:41:55 +08:00
|
|
|
"BatchNorm2DModule_basic",
|
|
|
|
"BatchNorm3DModule_basic",
|
2022-01-29 13:38:56 +08:00
|
|
|
"FlattenStaticModule_basic",
|
|
|
|
"FlattenRank0Module_basic",
|
|
|
|
"ElementwiseFlattenBroadcastModule_basic",
|
2022-01-30 01:10:50 +08:00
|
|
|
"SquareModule_basic",
|
2022-02-01 05:34:09 +08:00
|
|
|
"MaxPool2dStaticModule_basic",
|
2022-08-31 03:44:00 +08:00
|
|
|
"ResNet18StaticModule_basic",
|
2022-11-23 02:37:28 +08:00
|
|
|
"ReduceAmaxKeepDim_basic",
|
2022-02-04 06:08:19 +08:00
|
|
|
"NativeLayerNormModule4D_basic",
|
|
|
|
"LayerNormNormalizeOverAllDimsModule_basic",
|
|
|
|
"PermuteModule_basic",
|
|
|
|
"PermuteNegativeIndexModule_basic",
|
2022-02-12 04:30:02 +08:00
|
|
|
"ElementwiseLog2Module_basic",
|
|
|
|
"Threshold1dIntI32Module_basic",
|
|
|
|
"Threshold1dFloatModule_basic",
|
|
|
|
"Threshold2dFloatModule_basic",
|
|
|
|
"Threshold3dFloatModule_basic",
|
|
|
|
"ElementwiseSubScalarIntModule_basic",
|
|
|
|
"ElementwiseAddScalarIntModule_basic",
|
|
|
|
"ElementwiseMulScalarModule_basic",
|
2022-02-17 01:53:51 +08:00
|
|
|
"ZerosModuleDefaultDtype_basic",
|
|
|
|
"ZerosModuleInt2D_basic",
|
|
|
|
"ZerosModuleInt3D_basic",
|
|
|
|
"ZerosModuleFloat2D_basic",
|
|
|
|
"ZerosModuleFloat3D_basic",
|
|
|
|
"ZerosModuleFalsePinMemory_basic",
|
|
|
|
"OnesModuleDefaultDtype_basic",
|
|
|
|
"OnesModuleInt_basic",
|
|
|
|
"OnesModuleFloat_basic",
|
|
|
|
"OnesModuleFalsePinMemory_basic",
|
2022-08-19 14:53:21 +08:00
|
|
|
"OnesModuleCPUDevice_basic",
|
2022-02-28 14:14:40 +08:00
|
|
|
"NewZerosModuleDefaultDtype_basic",
|
|
|
|
"NewZerosModuleInt2D_basic",
|
|
|
|
"NewZerosModuleInt3D_basic",
|
|
|
|
"NewZerosModuleFloat2D_basic",
|
|
|
|
"NewZerosModuleFloat3D_basic",
|
|
|
|
"NewZerosModuleFalsePinMemory_basic",
|
|
|
|
"NewOnesModuleDefaultDtype_basic",
|
|
|
|
"NewOnesModuleInt2D_basic",
|
|
|
|
"NewOnesModuleInt3D_basic",
|
|
|
|
"NewOnesModuleFloat2D_basic",
|
|
|
|
"NewOnesModuleFloat3D_basic",
|
|
|
|
"NewOnesModuleFalsePinMemory_basic",
|
2022-03-02 01:30:58 +08:00
|
|
|
"SiluModule_basic",
|
2022-02-17 00:34:03 +08:00
|
|
|
"DropoutEvalIntModule_basic",
|
|
|
|
"DropoutEvalFloatModule_basic",
|
2022-03-26 05:15:07 +08:00
|
|
|
"ContiguousModule_basic",
|
|
|
|
"DropoutModule_basic",
|
|
|
|
"ViewExpandModule_basic",
|
2022-06-23 06:58:40 +08:00
|
|
|
"ViewExpandOnesModule_basic",
|
2022-07-01 07:39:25 +08:00
|
|
|
"ViewExpandOnesBeforeAndAfterModule_basic",
|
|
|
|
"ViewExpandOnesMiddleModule_basic",
|
2022-07-21 05:35:51 +08:00
|
|
|
"ViewExpandCollapseModule_basic",
|
|
|
|
"ViewExpandCollapseWithOnesModule_basic",
|
2022-11-24 21:02:59 +08:00
|
|
|
"ViewCollapseInferredDimModule_basic",
|
|
|
|
"ViewExpandInferredDimModule_basic",
|
2022-03-26 05:15:07 +08:00
|
|
|
"ViewNoChangeStaticModule_basic",
|
|
|
|
"UnsafeViewExpandModule_basic",
|
|
|
|
"ReshapeCollapseModule_basic",
|
2022-03-31 08:00:55 +08:00
|
|
|
"ElementwiseGeluModule_basic",
|
|
|
|
"GeluBackwardModule_basic",
|
2022-04-04 00:19:01 +08:00
|
|
|
"ElementwiseNeIntScalarModule_basic",
|
|
|
|
"ElementwiseNeFloatTensorModule_basic",
|
2022-07-08 14:44:03 +08:00
|
|
|
"Convolution2DStaticModule_basic",
|
2022-04-16 03:37:57 +08:00
|
|
|
"ElementwiseNegModule_basic",
|
2022-04-19 21:47:47 +08:00
|
|
|
"TestMultipleTensorReturn_basic",
|
2022-09-13 06:59:12 +08:00
|
|
|
"TypeAsSameModule_basic",
|
2022-05-19 20:35:59 +08:00
|
|
|
"AdaptiveAvgPool2dUnitOutputSizeStaticModule_basic",
|
2022-05-30 16:08:54 +08:00
|
|
|
"BaddbmmDynamicModule_basic",
|
|
|
|
"BaddbmmStaticModule_basic",
|
|
|
|
"BaddbmmWithAlphaBetaModule_basic",
|
|
|
|
"BaddbmmWithAlphaModule_basic",
|
|
|
|
"BaddbmmWithBetaModule_basic",
|
|
|
|
"BaddbmmBroadcast1DInputModule_basic",
|
|
|
|
"BaddbmmBroadcast2DInputModule_basic",
|
2022-06-03 20:38:59 +08:00
|
|
|
"NumpyTRank1Module_basic",
|
|
|
|
"NumpyTRank2Module_basic",
|
|
|
|
"NumpyTRankNStaticModule_basic",
|
|
|
|
"NumpyTRankNDynamicModule_basic",
|
2022-07-01 04:13:52 +08:00
|
|
|
"EmbeddingModuleI32Static_basic",
|
2022-07-08 04:05:33 +08:00
|
|
|
"TModuleRank2_basic",
|
|
|
|
"TransposeIntModule_basic",
|
|
|
|
"TransposeIntNegDimsModule_basic",
|
2022-07-14 01:10:18 +08:00
|
|
|
"ArgmaxModule_keepDim",
|
|
|
|
"ArgmaxModule_with_dim",
|
2022-08-11 08:02:06 +08:00
|
|
|
"_LogSoftmaxModuleStable_basic",
|
2022-10-19 00:39:39 +08:00
|
|
|
"ElementwiseAtenWhereSelfModule_basic",
|
2022-11-15 01:09:15 +08:00
|
|
|
"ElementwiseUnsqueezeBroadcastModule_basic",
|
2022-12-22 00:56:07 +08:00
|
|
|
"MaskedFillScalarIntValueStaticModule_basic",
|
|
|
|
"MaskedFillTensorIntValueStaticModule_basic",
|
2022-12-13 04:13:07 +08:00
|
|
|
"ElementwiseAddScalarInt64Module_basic",
|
|
|
|
"TensorLiteralModule_basic",
|
|
|
|
"TensorOpaqueLiteralModule_basic",
|
|
|
|
"TypePromotionDifferentCategoryModule_basic",
|
|
|
|
"TypePromotionSameCategoryDifferentWidthModule_basic",
|
|
|
|
"TypePromotionZeroRankHigherCategoryModule_basic",
|
2022-12-22 03:04:07 +08:00
|
|
|
"GatherStaticModule_basic",
|
2022-09-21 06:07:46 +08:00
|
|
|
"LiftFreshCopyModule_basic",
|
2022-09-21 02:38:09 +08:00
|
|
|
"ReduceSumDimIntListKeepDimNegativeDimStaticModule_basic",
|
2022-10-18 12:22:53 +08:00
|
|
|
"ReduceSumDimIntListFloatModule_basic",
|
|
|
|
"ReduceSumDimIntListIntModule_basic",
|
|
|
|
"ReduceSumDimIntListKeepDimFloatModule_basic",
|
|
|
|
"ReduceSumDimIntListKeepDimIntModule_basic",
|
|
|
|
"ReduceSumFloatModule_basic",
|
|
|
|
"ReduceSumSignedIntModule_basic",
|
|
|
|
"ReduceSumUnsignedIntModule_basic",
|
2022-09-30 00:40:56 +08:00
|
|
|
"BroadcastToSameRankStaticModule_basic",
|
|
|
|
"BroadcastZeroRankInputStaticModule_basic",
|
2022-09-23 16:23:09 +08:00
|
|
|
"SliceStaticModule_basic",
|
2022-09-30 22:33:41 +08:00
|
|
|
"ArangeStartStepIntModule_basic",
|
|
|
|
"ArangeDtypeFloatModule_basic",
|
|
|
|
"ArangeIntModule_basic",
|
|
|
|
"ArangeNegativeStartIntModule_basic",
|
|
|
|
"ArangeStartIntModule_basic",
|
|
|
|
"ArangeStartNegativeStepIntModule_basic",
|
|
|
|
"ArangeZeroElementOutputModule_basic",
|
2022-10-04 22:00:16 +08:00
|
|
|
"NumToTensorIntModule_basic",
|
2022-10-04 21:05:59 +08:00
|
|
|
"ToDtypeBoolLayoutNoneStaticModule_basic",
|
|
|
|
"ToCopyBoolDTypeStaticModule_basic",
|
|
|
|
"HardTanhIntModule_basic",
|
2022-10-20 19:02:09 +08:00
|
|
|
"AtenRoundIntModule_basic",
|
|
|
|
"MseLossNoReductionModule_basic",
|
2022-10-26 20:41:52 +08:00
|
|
|
"AddCMul_Module_basic",
|
|
|
|
"AddCDiv_Module_basic",
|
2022-12-01 23:46:17 +08:00
|
|
|
"TestF16Return_basic",
|
2021-10-26 02:43:21 +08:00
|
|
|
}
|
2022-06-10 03:56:01 +08:00
|
|
|
|
|
|
|
LTC_XFAIL_SET = {
|
2022-07-14 01:28:05 +08:00
|
|
|
"_Convolution2DAllFalseModule_basic",
|
|
|
|
"_Convolution2DBenchmarkModule_basic",
|
|
|
|
"_Convolution2DCudnnModule_basic",
|
|
|
|
"_Convolution2DDeterministicModule_basic",
|
|
|
|
"_Convolution2DTF32Module_basic",
|
2022-08-22 13:44:44 +08:00
|
|
|
"_ConvolutionDeprecated2DAllFalseModule_basic",
|
|
|
|
"_ConvolutionDeprecated2DBenchmarkModule_basic",
|
2022-09-27 00:16:49 +08:00
|
|
|
"_ConvolutionDeprecated2DCudnnModule_basic",
|
2022-08-22 13:44:44 +08:00
|
|
|
"_ConvolutionDeprecated2DDeterministicModule_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
"AdaptiveAvgPool2dNonUnitOutputSizeDynamicModule_basic",
|
|
|
|
"AdaptiveAvgPool2dNonUnitOutputSizeStaticModule_basic",
|
|
|
|
"AddIntModule_basic",
|
|
|
|
"BernoulliFloatModule_basic",
|
|
|
|
"BernoulliTensorModule_basic",
|
|
|
|
"BincountMinlengthModule_basic",
|
|
|
|
"BincountModule_basic",
|
|
|
|
"BincountStaticSizeModule_basic",
|
|
|
|
"BoolFloatFalseModule_basic",
|
|
|
|
"BoolFloatTrueModule_basic",
|
|
|
|
"BoolIntFalseModule_basic",
|
|
|
|
"BoolIntTrueModule_basic",
|
|
|
|
"CeilFloatModule_basic",
|
|
|
|
"DivFloatModule_basic",
|
2022-07-13 03:21:50 +08:00
|
|
|
"ElementwiseAtenFloorDivideBroadcastModule_basic",
|
|
|
|
"ElementwiseAtenFloorDivideModule_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
"EqIntModule_basic",
|
|
|
|
"GeFloatIntModule_basic",
|
|
|
|
"GeFloatModule_basic",
|
2022-07-30 17:54:40 +08:00
|
|
|
"GeIntModule_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
"GtFloatIntModule_basic",
|
|
|
|
"GtIntModule_basic",
|
|
|
|
"HBC_basic",
|
|
|
|
"IndexPut1DFloatAccumulateModule_basic",
|
|
|
|
"IndexPut1DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPut1DIntAccumulateModule_basic",
|
|
|
|
"IndexPut1DIntNonAccumulateModule_basic",
|
|
|
|
"IndexPut2DFloatAccumulateModule_basic",
|
|
|
|
"IndexPut2DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPut2DIntAccumulateModule_basic",
|
|
|
|
"IndexPut2DIntNonAccumulateModule_basic",
|
|
|
|
"IndexPut3DFloatAccumulateModule_basic",
|
|
|
|
"IndexPut3DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPut3DIntAccumulateModule_basic",
|
|
|
|
"IndexPut3DIntNonAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin1DFloatAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin1DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin1DIntAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin1DIntNonAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin2DFloatAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin2DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin2DIntAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin2DIntNonAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin3DFloatAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin3DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin3DIntAccumulateModule_basic",
|
|
|
|
"IndexPutHackedTwin3DIntNonAccumulateModule_basic",
|
|
|
|
"IndexPutImpl1DFloatAccumulateModule_basic",
|
|
|
|
"IndexPutImpl1DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPutImpl1DIntAccumulateModule_basic",
|
|
|
|
"IndexPutImpl1DIntNonAccumulateModule_basic",
|
|
|
|
"IndexPutImpl2DFloatAccumulateModule_basic",
|
|
|
|
"IndexPutImpl2DFloatNonAccumulateModule_basic",
|
|
|
|
"IndexPutImpl3DFloatAccumulateModule_basic",
|
|
|
|
"IndexPutImpl3DFloatNonAccumulateModule_basic",
|
2022-07-13 03:21:50 +08:00
|
|
|
"IndexTensorModule3dInput_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
"IndexTensorModule_basic",
|
2022-07-30 17:54:40 +08:00
|
|
|
"IndexTensorMultiInputContiguousCenter_basic",
|
|
|
|
"IndexTensorMultiInputNonContiguous_basic",
|
|
|
|
"IndexTensorMultiInputOneDim_basic",
|
|
|
|
"IndexTensorMultiInputThreeIndexers_basic",
|
|
|
|
"IndexTensorMultiInput_basic",
|
2022-07-26 07:40:25 +08:00
|
|
|
"IndexTensorSelectDimModule_basic",
|
2022-08-16 20:46:42 +08:00
|
|
|
"IndexTensorMultiInputContiguousOneDimDynamic_basic",
|
|
|
|
"IndexTensorMultiInputNonContiguousOneDimDynamic_basic",
|
|
|
|
"IndexTensorMultiInputNonContiguousDynamic_basic",
|
|
|
|
"IndexTensorMultiInputNonContiguousMultipleStaticDims_basic",
|
2022-09-06 21:29:24 +08:00
|
|
|
"IndexTensorHackedTwinModule_basic",
|
|
|
|
"IndexTensorHackedTwinModule3dInput_basic",
|
|
|
|
"IndexTensorHackedTwinMultiInputNonContiguousMultipleStaticDims_basic",
|
2022-09-27 00:16:49 +08:00
|
|
|
"LiftFreshCopyModule_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
"Matmul_dot",
|
|
|
|
"MulIntModule_basic",
|
2022-10-06 21:11:52 +08:00
|
|
|
"DivIntModule_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
"NeFloatIntModule_basic",
|
|
|
|
"NeIntModule_basic",
|
|
|
|
"QuantizedMLP_basic",
|
|
|
|
"RandLikeDtypeModule_basic",
|
|
|
|
"RandLikeModule_basic",
|
2022-08-24 08:36:05 +08:00
|
|
|
"RollModule_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
"ScalarImplicitFloatModule_basic",
|
|
|
|
"ScalarImplicitIntModule_basic",
|
|
|
|
"SliceEndSleStartModule_basic",
|
|
|
|
"SliceOutOfUpperBoundIndexModule_basic",
|
|
|
|
"SliceStartEqEndModule_basic",
|
|
|
|
"SqrtIntModule_basic",
|
|
|
|
"SubFloatModule_basic",
|
|
|
|
"SubIntModule_basic",
|
2022-07-26 07:40:25 +08:00
|
|
|
"TensorsConcatNegativeDimModule_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
"TensorToBoolZeroRank_basic",
|
|
|
|
"TensorToBool_basic",
|
|
|
|
"TensorToFloatZeroRank_basic",
|
|
|
|
"TensorToFloat_basic",
|
|
|
|
"TensorToIntZeroRank_basic",
|
|
|
|
"TensorToInt_basic",
|
|
|
|
"TensorsConcatModule_basic",
|
|
|
|
"UniformModule_basic",
|
2022-12-02 00:30:10 +08:00
|
|
|
"UniformNoCorrelationModule_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
"UnsafeViewCollapseDynamicWithAtenSizeIntModule_basic",
|
|
|
|
"ViewCollapseDynamicWithAtenSizeIntModule_basic",
|
2022-08-02 06:23:45 +08:00
|
|
|
"AtenEmbeddingBagSumExample_basic",
|
2022-08-09 06:56:49 +08:00
|
|
|
"Aten_EmbeddingBagExample_basic",
|
2022-08-11 08:02:06 +08:00
|
|
|
"ElementwiseRemainderScalarModule_Int_Float_basic",
|
|
|
|
"ElementwiseRemainderScalarModule_Float_basic",
|
|
|
|
"ElementwiseRemainderScalarModule_Int_basic",
|
|
|
|
"ElementwiseRemainderScalarModule_Bool_basic",
|
2022-09-20 02:50:51 +08:00
|
|
|
"AtenIntTensorByteDtypeModule_basic",
|
|
|
|
"AtenIntTensorCharDtypeModule_basic",
|
2022-12-22 00:56:07 +08:00
|
|
|
"MaskedFillTensorIntValueStaticModule_basic",
|
2022-10-26 20:18:49 +08:00
|
|
|
"Fill_TensorFloat32WithFloat32_basic",
|
|
|
|
"Fill_TensorFloat32WithFloat64_basic",
|
|
|
|
"Fill_TensorFloat32WithInt64_basic",
|
2022-11-01 21:08:04 +08:00
|
|
|
"UpSampleNearest2dBackwardVec_basic",
|
|
|
|
"UpSampleNearest2dBackwardOutputSizeNone_basic",
|
2022-11-04 15:57:29 +08:00
|
|
|
"ConvolutionBackwardModule2D_basic",
|
|
|
|
"ConvolutionBackwardModule2DPadded_basic",
|
2022-11-15 22:39:40 +08:00
|
|
|
"VarMeanCorrectionModule_basic",
|
2022-11-21 16:38:47 +08:00
|
|
|
"VarMeanCorrectionNoneModule_basic",
|
|
|
|
"PrimsConvertElementTypeModule_basic",
|
2022-11-29 21:05:58 +08:00
|
|
|
"CopyModule_basic",
|
|
|
|
"CopyWithDifferentDTypesAndSizesModule_basic",
|
|
|
|
"CopyWithDifferentDTypesModule_basic",
|
|
|
|
"CopyWithDifferentSizesModule_basic",
|
2022-12-09 21:59:46 +08:00
|
|
|
"Conv2dNoPaddingModule_basic",
|
|
|
|
"Conv2dWithPaddingDilationStrideModule_basic",
|
|
|
|
"Conv2dWithPaddingDilationStrideStaticModule_basic",
|
|
|
|
"Conv2dWithPaddingModule_basic",
|
|
|
|
"Conv_Transpose2dModule_basic",
|
|
|
|
"Convolution2DModule_basic",
|
|
|
|
"Convolution2DStaticModule_basic",
|
|
|
|
"Convolution2DStridedModule_basic",
|
|
|
|
"ConvolutionModule2DGroups_basic",
|
|
|
|
"ConvolutionModule2DTransposeStridedStatic_basic",
|
|
|
|
"ConvolutionModule2DTransposeStrided_basic",
|
|
|
|
"ConvolutionModule2DTranspose_basic",
|
|
|
|
"ElementwiseClampModule_basic",
|
2022-12-28 11:21:33 +08:00
|
|
|
"ElementwisePreluModule_basic",
|
2022-12-09 21:59:46 +08:00
|
|
|
"IouOfModule_basic",
|
|
|
|
"MobilenetV3Module_basic",
|
|
|
|
"NativeBatchNormNoneWeightModule_basic",
|
|
|
|
"NllLossModuleBackward1DMean_basic",
|
|
|
|
"NllLossModuleBackward1DSum_basic",
|
|
|
|
"NllLossModuleBackward1D_basic",
|
|
|
|
"NllLossModuleBackwardMean_basic",
|
|
|
|
"NllLossModuleBackwardSum_basic",
|
|
|
|
"NllLossModuleBackward_basic",
|
|
|
|
"NllLossModuleBackward_ignore_index",
|
|
|
|
"NllLossModule_1D_basic",
|
|
|
|
"NllLossModule_basic",
|
|
|
|
"NllLossModule_ignore_index_out_of_bounds_basic",
|
|
|
|
"NllLossModule_mean_basic",
|
|
|
|
"NllLossModule_sum_basic",
|
|
|
|
"ResNet18Module_basic",
|
|
|
|
"ResNet18StaticModule_basic",
|
2022-12-09 23:22:26 +08:00
|
|
|
"VarMeanBiasedModule_basic",
|
|
|
|
"VarMeanUnbiasedModule_basic",
|
2022-06-10 03:56:01 +08:00
|
|
|
}
|