mirror of https://github.com/llvm/torch-mlir
[FxImporter][TOSA] Enable FxImporter to TOSA e2e tests (#3349)
Signed-off-by: Suraj Sudhir <suraj.sudhir@arm.com>pull/3338/head
Suraj Sudhir
GitHub
parent
ccb772cd0f
commit
0ca88028cd
|
|
@ -55,6 +55,7 @@ from .xfail_sets import (
|
||||||
FX_IMPORTER_CRASHING_SET,
|
FX_IMPORTER_CRASHING_SET,
|
||||||
FX_IMPORTER_STABLEHLO_XFAIL_SET,
|
FX_IMPORTER_STABLEHLO_XFAIL_SET,
|
||||||
FX_IMPORTER_STABLEHLO_CRASHING_SET,
|
FX_IMPORTER_STABLEHLO_CRASHING_SET,
|
||||||
|
FX_IMPORTER_TOSA_XFAIL_SET,
|
||||||
)
|
)
|
||||||
|
|
||||||
# Import tests to register them in the global registry.
|
# Import tests to register them in the global registry.
|
||||||
|
|
@ -76,6 +77,7 @@ def _get_argparse():
|
||||||
"onnx",
|
"onnx",
|
||||||
"fx_importer",
|
"fx_importer",
|
||||||
"fx_importer_stablehlo",
|
"fx_importer_stablehlo",
|
||||||
|
"fx_importer_tosa",
|
||||||
]
|
]
|
||||||
parser = argparse.ArgumentParser(description="Run torchscript e2e tests.")
|
parser = argparse.ArgumentParser(description="Run torchscript e2e tests.")
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
|
|
@ -95,6 +97,7 @@ Meaning of options:
|
||||||
"onnx": export to the model via onnx and reimport using the torch-onnx-to-torch path.
|
"onnx": export to the model via onnx and reimport using the torch-onnx-to-torch path.
|
||||||
"fx_importer": run the model through the fx importer frontend and execute the graph using Linalg-on-Tensors.
|
"fx_importer": run the model through the fx importer frontend and execute the graph using Linalg-on-Tensors.
|
||||||
"fx_importer_stablehlo": run the model through the fx importer frontend and execute the graph using Stablehlo backend.
|
"fx_importer_stablehlo": run the model through the fx importer frontend and execute the graph using Stablehlo backend.
|
||||||
|
"fx_importer_tosa": run the model through the fx importer frontend and execute the graph using the TOSA backend.
|
||||||
""",
|
""",
|
||||||
)
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
|
|
@ -179,6 +182,10 @@ def main():
|
||||||
config = FxImporterTestConfig(LinalgOnTensorsStablehloBackend(), "stablehlo")
|
config = FxImporterTestConfig(LinalgOnTensorsStablehloBackend(), "stablehlo")
|
||||||
xfail_set = FX_IMPORTER_STABLEHLO_XFAIL_SET
|
xfail_set = FX_IMPORTER_STABLEHLO_XFAIL_SET
|
||||||
crashing_set = FX_IMPORTER_STABLEHLO_CRASHING_SET
|
crashing_set = FX_IMPORTER_STABLEHLO_CRASHING_SET
|
||||||
|
elif args.config == "fx_importer_tosa":
|
||||||
|
config = FxImporterTestConfig(LinalgOnTensorsTosaBackend(), "tosa")
|
||||||
|
xfail_set = FX_IMPORTER_TOSA_XFAIL_SET
|
||||||
|
crashing_set = set()
|
||||||
elif args.config == "torchdynamo":
|
elif args.config == "torchdynamo":
|
||||||
config = TorchDynamoTestConfig(RefBackendLinalgOnTensorsBackend())
|
config = TorchDynamoTestConfig(RefBackendLinalgOnTensorsBackend())
|
||||||
xfail_set = TORCHDYNAMO_XFAIL_SET
|
xfail_set = TORCHDYNAMO_XFAIL_SET
|
||||||
|
|
|
||||||
|
|
@ -2690,3 +2690,749 @@ ONNX_CRASHING_SET = {
|
||||||
# For now, we are removing the test until this issue has been debugged.
|
# For now, we are removing the test until this issue has been debugged.
|
||||||
"QuantizedMLP_basic",
|
"QuantizedMLP_basic",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
FX_IMPORTER_TOSA_XFAIL_SET = {
|
||||||
|
"AdaptiveAvgPool1dGeneralDynamicNoBatches_basic",
|
||||||
|
"AdaptiveAvgPool1dGeneralDynamic_basic",
|
||||||
|
"AdaptiveAvgPool1dStaticLargerOutput_basic",
|
||||||
|
"AdaptiveAvgPool2dDynamicNoBatch_basic",
|
||||||
|
"AdaptiveAvgPool2dDynamic_basic",
|
||||||
|
"AdaptiveAvgPool3dDynamicNoBatch_basic",
|
||||||
|
"AdaptiveAvgPool3dDynamic_basic",
|
||||||
|
"AdaptiveMaxPool1dDynamicNoBatch_basic",
|
||||||
|
"AdaptiveMaxPool1dDynamic_basic",
|
||||||
|
"AdaptiveMaxPool1dStatic_basic",
|
||||||
|
"AdaptiveMaxPool2dDynamicNoBatch_basic",
|
||||||
|
"AdaptiveMaxPool2dDynamicWithIndices_basic",
|
||||||
|
"AdaptiveMaxPool2dDynamic_basic",
|
||||||
|
"AdaptiveMaxPool2dStaticWithIndices_basic",
|
||||||
|
"AdaptiveMaxPool2dStatic_basic",
|
||||||
|
"AdaptiveMaxPool3dDynamicNoBatch_basic",
|
||||||
|
"AdaptiveMaxPool3dDynamicWithIndices_basic",
|
||||||
|
"AdaptiveMaxPool3dDynamic_basic",
|
||||||
|
"AdaptiveMaxPool3dStaticWithIndices_basic",
|
||||||
|
"AdaptiveMaxPool3dStatic_basic",
|
||||||
|
"AddIntModule_basic",
|
||||||
|
"Add_MixPModule_basic",
|
||||||
|
"AllBoolFalseModule_basic",
|
||||||
|
"AllBoolTrueModule_basic",
|
||||||
|
"AnyBoolFalseModule_basic",
|
||||||
|
"AnyBoolTrueModule_basic",
|
||||||
|
"ArangeStartOutViewModule_basic",
|
||||||
|
"ArgminIntModule_basic",
|
||||||
|
"ArgminIntModule_multiple_mins",
|
||||||
|
"ArgminModule_basic",
|
||||||
|
"ArgminModule_keepDim",
|
||||||
|
"ArgminModule_with_dim",
|
||||||
|
"AtenComplexImagModule_basic",
|
||||||
|
"AtenComplexRealModule_basic",
|
||||||
|
"AtenComplexViewModule_basic",
|
||||||
|
"AtenDiagEmbedDefaultDiag_basic",
|
||||||
|
"AtenDiagEmbedDimDiag_basic",
|
||||||
|
"AtenDiagEmbedNegOffsetDiag_basic",
|
||||||
|
"AtenDiagEmbedNonDefault4DDiag_basic",
|
||||||
|
"AtenDiagEmbedOffsetDiag_basic",
|
||||||
|
"AtenDiagEmbedRevDimDiag_basic",
|
||||||
|
"AtenEmbeddingBagStaticModule_basic",
|
||||||
|
"AtenEmbeddingBagSumExample_basic",
|
||||||
|
"AtenEyeMModuleInt2D_basic",
|
||||||
|
"AtenEyeModuleInt2D_basic",
|
||||||
|
"AtenFloatScalarModule_basic",
|
||||||
|
"AtenInstanceNormModule_basic",
|
||||||
|
"AtenIntBoolOpConstFalseModule_basic",
|
||||||
|
"AtenIntBoolOpConstTrueModule_basic",
|
||||||
|
"AtenIntBoolOpModule_basic",
|
||||||
|
"AtenIntTensorByteDtypeModule_basic",
|
||||||
|
"AtenIntTensorCharDtypeModule_basic",
|
||||||
|
"AtenItemFpOpModule_basic",
|
||||||
|
"AtenItemIntOpModule_basic",
|
||||||
|
"AtenLinalgCrossBroadcast_basic",
|
||||||
|
"AtenLinalgCrossCustomDim_basic",
|
||||||
|
"AtenLinalgCrossDynamic_basic",
|
||||||
|
"AtenLinalgCrossFloat_basic",
|
||||||
|
"AtenLinalgCrossInt_basic",
|
||||||
|
"AtenLinalgCrossNegativeDim_basic",
|
||||||
|
"AtenMatmulQMixedSigni8Transpose_basic",
|
||||||
|
"AtenMatmulQMixedSigni8_basic",
|
||||||
|
"AtenMatmulQint8MV_basic",
|
||||||
|
"AtenMatmulQint8VM_basic",
|
||||||
|
"AtenMatmulQint8VV_basic",
|
||||||
|
"AtenMatmulQint8_basic",
|
||||||
|
"AtenMmIntTypes_basic",
|
||||||
|
"AtenMmQMixedSigni8_basic",
|
||||||
|
"AtenMmQint8_basic",
|
||||||
|
"AtenMmQuint8_basic",
|
||||||
|
"AtenRealView128Module_basic",
|
||||||
|
"AtenRealView64Module_basic",
|
||||||
|
"AtenRoundFloatHalfToEvenModule_basic",
|
||||||
|
"AtenRoundFloatModule_basic",
|
||||||
|
"AtenSubFloatModule_basic",
|
||||||
|
"AtenTopKModule_basic",
|
||||||
|
"AtenTopKSmallestModule_basic",
|
||||||
|
"AtenTrilModule_basic",
|
||||||
|
"AtenTrilWithNegDiagonalModule_basic",
|
||||||
|
"AtenTrilWithPosDiagonalModule_basic",
|
||||||
|
"Aten_CastLongModule_basic",
|
||||||
|
"Aten_EmbeddingBagExample_basic",
|
||||||
|
"AvgPool1dFloatModule_basic",
|
||||||
|
"AvgPool1dIntModule_basic",
|
||||||
|
"AvgPool1dStaticModule_basic",
|
||||||
|
"AvgPool2dCeilModeTrueModule_basic",
|
||||||
|
"AvgPool2dDivisorOverrideModule_basic",
|
||||||
|
"AvgPool2dFloatModule_basic",
|
||||||
|
"AvgPool2dIntModule_basic",
|
||||||
|
"AvgPool2dStaticModule_basic",
|
||||||
|
"BernoulliFloatModule_basic",
|
||||||
|
"BernoulliModule_basic",
|
||||||
|
"BernoulliOnesModule_basic",
|
||||||
|
"BernoulliPModule_basic",
|
||||||
|
"BernoulliTensorModule_basic",
|
||||||
|
"BernoulliZerosModule_basic",
|
||||||
|
"BincountMinlengthModule_basic",
|
||||||
|
"BincountModule_basic",
|
||||||
|
"BincountStaticSizeModule_basic",
|
||||||
|
"BmmIntModule_basic",
|
||||||
|
"BoolFloatConstantModule_basic",
|
||||||
|
"BoolFloatFalseModule_basic",
|
||||||
|
"BoolFloatTrueModule_basic",
|
||||||
|
"BoolIntConstantModule_basic",
|
||||||
|
"BoolIntFalseModule_basic",
|
||||||
|
"BoolIntTrueModule_basic",
|
||||||
|
"BroadcastDynamicDimModule_basic",
|
||||||
|
"BroadcastToModule_basic",
|
||||||
|
"CeilFloatModule_basic",
|
||||||
|
"CollapseAllDimensionsModule_basic",
|
||||||
|
"CollapseFullDynamicModule_basic",
|
||||||
|
"CollapsePartialDynamicModule_basic",
|
||||||
|
"CollapseRank1DynamicModule_basic",
|
||||||
|
"CollapseStaticModule_basic",
|
||||||
|
"ConstantBoolParameterModule_basic",
|
||||||
|
"ContainsIntList_False",
|
||||||
|
"ContainsIntList_True",
|
||||||
|
"Conv1dModule_basic",
|
||||||
|
"Conv2dQInt8Module_basic",
|
||||||
|
"Conv2dWithPaddingDilationStrideStaticModule_grouped",
|
||||||
|
"Conv2dWithPaddingDilationStrideStaticModule_grouped_multiplier",
|
||||||
|
"Conv3dModule_basic",
|
||||||
|
"ConvTbcModule_basic",
|
||||||
|
"ConvTranspose2DQInt8_basic",
|
||||||
|
"Conv_Transpose2dModule_basic",
|
||||||
|
"ConvolutionBackwardModule2DPadded_basic",
|
||||||
|
"ConvolutionBackwardModule2DStatic_basic",
|
||||||
|
"ConvolutionBackwardModule2DStrided_basic",
|
||||||
|
"ConvolutionBackwardModule2D_basic",
|
||||||
|
"ConvolutionModule2DGroups_basic",
|
||||||
|
"ConvolutionModule2DTransposeNonUnitOutputPadding_basic",
|
||||||
|
"ConvolutionModule2DTransposeStridedStatic_basic",
|
||||||
|
"ConvolutionModule2DTransposeStrided_basic",
|
||||||
|
"ConvolutionModule2DTranspose_basic",
|
||||||
|
"CopyWithDifferentDTypesModule_basic",
|
||||||
|
"CosineSimilarityStaticBroadcastModule_basic",
|
||||||
|
"CrossEntropyLossModule_basic",
|
||||||
|
"CumsumInputDtypeInt32Module_basic",
|
||||||
|
"CumsumModule_basic",
|
||||||
|
"CumsumStaticModule_basic",
|
||||||
|
"CumsumStaticNegativeDimModule_basic",
|
||||||
|
"DiagonalModule_basic",
|
||||||
|
"DiagonalModule_nonsquare",
|
||||||
|
"DiagonalModule_transposed",
|
||||||
|
"DiagonalModule_with_dims",
|
||||||
|
"DiagonalModule_with_dims_and_offset",
|
||||||
|
"DiagonalModule_with_negative_dims",
|
||||||
|
"DiagonalModule_with_offset",
|
||||||
|
"DiagonalWithStaticShapeModule_basic",
|
||||||
|
"DivFloatModule_basic",
|
||||||
|
"DivIntModule_basic",
|
||||||
|
"DropoutTrainModule_basic",
|
||||||
|
"DropoutTrainStaticShapeModule_basic",
|
||||||
|
"ElementwiseAcosIntModule_basic",
|
||||||
|
"ElementwiseAcosModule_basic",
|
||||||
|
"ElementwiseAcoshIntModule_basic",
|
||||||
|
"ElementwiseAcoshModule_basic",
|
||||||
|
"ElementwiseAddScalar_NumToTensorFloat_Module_basic",
|
||||||
|
"ElementwiseAndScalarModule_basic",
|
||||||
|
"ElementwiseAndScalarStaticShapeModule_basic",
|
||||||
|
"ElementwiseAsinIntModule_basic",
|
||||||
|
"ElementwiseAsinModule_basic",
|
||||||
|
"ElementwiseAsinhIntModule_basic",
|
||||||
|
"ElementwiseAsinhModule_basic",
|
||||||
|
"ElementwiseAtan2FloatIntModule_basic",
|
||||||
|
"ElementwiseAtan2FloatIntStaticModule_basic",
|
||||||
|
"ElementwiseAtan2TensorFloatModule_basic",
|
||||||
|
"ElementwiseAtan2TensorFloatStaticModule_basic",
|
||||||
|
"ElementwiseAtan2TensorIntModule_basic",
|
||||||
|
"ElementwiseAtan2TensorIntStaticModule_basic",
|
||||||
|
"ElementwiseAtanTensorFloatModule_basic",
|
||||||
|
"ElementwiseAtanTensorIntModule_basic",
|
||||||
|
"ElementwiseAtanhIntModule_basic",
|
||||||
|
"ElementwiseAtanhModule_basic",
|
||||||
|
"ElementwiseAtenFloorDivideBroadcastModule_basic",
|
||||||
|
"ElementwiseAtenFloorDivideScalarModule_basic",
|
||||||
|
"ElementwiseAtenFloorDivideScalarNegativeModule_basic",
|
||||||
|
"ElementwiseAtenFloorDivideTensorNegativeModule_basic",
|
||||||
|
"ElementwiseAtenFloorDivideTensorPositiveModule_basic",
|
||||||
|
"ElementwiseAtenLogicalAndOpModule_basic",
|
||||||
|
"ElementwiseAtenLogicalAndOpPromoteBroadcastModule_basic",
|
||||||
|
"ElementwiseAtenLogicalAndOpPromoteBroadcastStaticShapeModule_basic",
|
||||||
|
"ElementwiseAtenLogicalNotOpModule_basic",
|
||||||
|
"ElementwiseAtenLogicalNotOpPromoteModule_basic",
|
||||||
|
"ElementwiseAtenLogicalXorOpModule_basic",
|
||||||
|
"ElementwiseAtenLogicalXorOpPromoteBroadcastModule_basic",
|
||||||
|
"ElementwiseAtenLogicalXorOpPromoteBroadcastStaticShapeModule_basic",
|
||||||
|
"ElementwiseBitwiseAndScalarInt32Module_basic",
|
||||||
|
"ElementwiseBitwiseAndScalarInt64Module_basic",
|
||||||
|
"ElementwiseBitwiseAndScalarInt8Module_basic",
|
||||||
|
"ElementwiseBitwiseLeftShiftInt32Module_basic",
|
||||||
|
"ElementwiseBitwiseLeftShiftInt64Module_basic",
|
||||||
|
"ElementwiseBitwiseLeftShiftInt8Module_basic",
|
||||||
|
"ElementwiseBitwiseRightShiftInt32Module_basic",
|
||||||
|
"ElementwiseBitwiseRightShiftInt64Module_basic",
|
||||||
|
"ElementwiseBitwiseRightShiftInt8Module_basic",
|
||||||
|
"ElementwiseClampMinTensorFloatModule_basic",
|
||||||
|
"ElementwiseClampMinTensorIntModule_basic",
|
||||||
|
"ElementwiseClampTensorFloatModule_basic",
|
||||||
|
"ElementwiseClampTensorIntModule_basic",
|
||||||
|
"ElementwiseCosIntModule_basic",
|
||||||
|
"ElementwiseCosModule_basic",
|
||||||
|
"ElementwiseCoshIntModule_basic",
|
||||||
|
"ElementwiseCoshModule_basic",
|
||||||
|
"ElementwiseDequantizePerChannelModule_basic",
|
||||||
|
"ElementwiseDequantizePerTensorModule_basic",
|
||||||
|
"ElementwiseDivScalarRoundingModeFloorIntStaticModule_basic",
|
||||||
|
"ElementwiseDivScalarRoundingModeFloorModule_basic",
|
||||||
|
"ElementwiseDivScalarRoundingModeFloorStaticModule_basic",
|
||||||
|
"ElementwiseDivScalarRoundingModeTruncIntStaticModule_basic",
|
||||||
|
"ElementwiseDivScalarRoundingModeTruncModule_basic",
|
||||||
|
"ElementwiseDivScalarRoundingModeTruncStaticModule_basic",
|
||||||
|
"ElementwiseDivTensorFloatModule_basic",
|
||||||
|
"ElementwiseDivTensorRoundingModeFloorIntStaticModule_basic",
|
||||||
|
"ElementwiseDivTensorRoundingModeFloorModule_basic",
|
||||||
|
"ElementwiseDivTensorRoundingModeFloorStaticModule_basic",
|
||||||
|
"ElementwiseDivTensorRoundingModeTruncIntStaticModule_basic",
|
||||||
|
"ElementwiseDivTensorRoundingModeTruncModule_basic",
|
||||||
|
"ElementwiseDivTensorRoundingModeTruncStaticModule_basic",
|
||||||
|
"ElementwiseErfIntModule_basic",
|
||||||
|
"ElementwiseErfModule_basic",
|
||||||
|
"ElementwiseExpIntModule_basic",
|
||||||
|
"ElementwiseExpm1IntModule_basic",
|
||||||
|
"ElementwiseExpm1Module_basic",
|
||||||
|
"ElementwiseFmodTensor_Float_basic",
|
||||||
|
"ElementwiseFmodTensor_Int_Float_basic",
|
||||||
|
"ElementwiseFmodTensor_Int_basic",
|
||||||
|
"ElementwiseGeFloatTensorModule_basic",
|
||||||
|
"ElementwiseGeIntTensorModule_basic",
|
||||||
|
"ElementwiseGeluApproximateTanhModule_basic",
|
||||||
|
"ElementwiseHardshrinkModule_basic",
|
||||||
|
"ElementwiseHardshrinkStaticModule_basic",
|
||||||
|
"ElementwiseIntTensorLtFloatScalarModule_basic",
|
||||||
|
"ElementwiseLeFloatIntScalarModule_basic",
|
||||||
|
"ElementwiseLeFloatScalarModule_basic",
|
||||||
|
"ElementwiseLeFloatTensorNanModule_basic",
|
||||||
|
"ElementwiseLeIntScalarModule_basic",
|
||||||
|
"ElementwiseLeMixedIntScalarModule_basic",
|
||||||
|
"ElementwiseLog10IntModule_basic",
|
||||||
|
"ElementwiseLog10Module_basic",
|
||||||
|
"ElementwiseLog1pModule_basic",
|
||||||
|
"ElementwiseLog2IntModule_basic",
|
||||||
|
"ElementwiseLogIntModule_basic",
|
||||||
|
"ElementwiseLogSigmoidModule_basic",
|
||||||
|
"ElementwiseLogitModule_basic",
|
||||||
|
"ElementwiseMishModule_basic",
|
||||||
|
"ElementwiseMulTensorComplexDiffModule_basic",
|
||||||
|
"ElementwiseMulTensorComplexModule_basic",
|
||||||
|
"ElementwiseMulTensorFloatModule_basic",
|
||||||
|
"ElementwisePowScalarModule_basic",
|
||||||
|
"ElementwisePowTensorBroadcastModule_basic",
|
||||||
|
"ElementwisePowTensorBroadcastStaticModule_basic",
|
||||||
|
"ElementwisePowTensorModule_basic",
|
||||||
|
"ElementwisePowTensorStaticModule_basic",
|
||||||
|
"ElementwiseQuantizePerTensorModule_basic",
|
||||||
|
"ElementwiseQuantizePerTensorUIntModule_basic",
|
||||||
|
"ElementwiseReciprocalIntModule_basic",
|
||||||
|
"ElementwiseRemainderScalarModule_Bool_basic",
|
||||||
|
"ElementwiseRemainderTensorModule_Float_basic",
|
||||||
|
"ElementwiseRemainderTensorModule_Int_Float_basic",
|
||||||
|
"ElementwiseRemainderTensorModule_Int_basic",
|
||||||
|
"ElementwiseRsqrtIntModule_basic",
|
||||||
|
"ElementwiseSigmoidIntModule_basic",
|
||||||
|
"ElementwiseSinIntModule_basic",
|
||||||
|
"ElementwiseSinModule_basic",
|
||||||
|
"ElementwiseSinhIntModule_basic",
|
||||||
|
"ElementwiseSinhModule_basic",
|
||||||
|
"ElementwiseTanIntModule_basic",
|
||||||
|
"ElementwiseTanModule_basic",
|
||||||
|
"ElementwiseTernaryModule_basic",
|
||||||
|
"ElementwiseToDtypeF32ToI64Module_basic",
|
||||||
|
"ElementwiseToDtypeI64ToUI8Module_basic",
|
||||||
|
"ElementwiseUnaryIntModule_basic",
|
||||||
|
"ElementwiseWhereScalarOtherModule_basic",
|
||||||
|
"ElementwiseWhereScalarOtherStaticModule_basic",
|
||||||
|
"ElementwiseWhereScalarSelfModule_basic",
|
||||||
|
"ElementwiseWhereScalarSelfStaticModule_basic",
|
||||||
|
"EmptyLikeMemoryFormatModule_basic",
|
||||||
|
"EmptyLikeModule_defaultDtype",
|
||||||
|
"EmptyLikeModule_falsePinMemory",
|
||||||
|
"EmptyLikeModule_float",
|
||||||
|
"EmptyLikeModule_int",
|
||||||
|
"EmptyModule_contiguous",
|
||||||
|
"EmptyModule_defaultDtype",
|
||||||
|
"EmptyModule_falsePinMemory",
|
||||||
|
"EmptyModule_float",
|
||||||
|
"EmptyModule_int",
|
||||||
|
"EmptyModule_uint8",
|
||||||
|
"EmptyStridedModule_basic",
|
||||||
|
"EmptyStridedSizeIntStrideModule_basic",
|
||||||
|
"EqIntModule_basic",
|
||||||
|
"ExpandModule_basic",
|
||||||
|
"ExponentialModule_basic",
|
||||||
|
"FakeQuantizePerTensorAffineDynamicShapeModule_basic",
|
||||||
|
"FakeQuantizePerTensorAffineModule_basic",
|
||||||
|
"FakeQuantizePerTensorAffineRoundToEvenModule_basic",
|
||||||
|
"Fill_TensorFloat32WithFloat32_basic",
|
||||||
|
"Fill_TensorFloat32WithFloat64_basic",
|
||||||
|
"Fill_TensorFloat32WithInt64_basic",
|
||||||
|
"Fill_TensorFloat64WithFloat32Static_basic",
|
||||||
|
"Fill_TensorFloat64WithFloat32_basic",
|
||||||
|
"Fill_TensorFloat64WithFloat64_basic",
|
||||||
|
"Fill_TensorFloat64WithInt64Static_basic",
|
||||||
|
"Fill_TensorFloat64WithInt64_basic",
|
||||||
|
"FlipModuleStaticShape_basic",
|
||||||
|
"FlipModule_basic",
|
||||||
|
"FlipNegativeIndexModule_basic",
|
||||||
|
"FloatImplicitModule_basic",
|
||||||
|
"FullLikeModuleInt2D_basic",
|
||||||
|
"FullLikeModuleInt3D_basic",
|
||||||
|
"FullModuleDefaultDtype_basic",
|
||||||
|
"FullModuleFalsePinMemory_basic",
|
||||||
|
"FullModuleFloat2D_basic",
|
||||||
|
"FullModuleFloat3D_basic",
|
||||||
|
"FullModuleInt2D_basic",
|
||||||
|
"FullModuleInt3D_basic",
|
||||||
|
"GeFloatIntModule_basic",
|
||||||
|
"GeFloatModule_basic",
|
||||||
|
"GeIntModule_basic",
|
||||||
|
"GridSamplerBasic1_basic",
|
||||||
|
"GridSamplerBasic2_basic",
|
||||||
|
"GridSamplerBasic3_basic",
|
||||||
|
"GridSamplerBasic4_basic",
|
||||||
|
"GroupNormModule_basic",
|
||||||
|
"GroupNormNoWeightAndBiasModule_basic",
|
||||||
|
"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",
|
||||||
|
"IndexPutImpl2DImplicitModule_basic",
|
||||||
|
"IndexPutImpl2DIndexModule_basic",
|
||||||
|
"IndexPutImpl2DNoneIndexStaticModule_basic",
|
||||||
|
"IndexPutImpl3DFloatAccumulateModule_basic",
|
||||||
|
"IndexPutImpl3DFloatNonAccumulateModule_basic",
|
||||||
|
"IndexPutImplIndexWithNoneModule_basic",
|
||||||
|
"IndexSelectDynamicIndexSizeModule_basic",
|
||||||
|
"IndexSelectDynamicInputSizeModule_basic",
|
||||||
|
"IndexSelectDynamicModulebasic",
|
||||||
|
"IndexSelectNegativeDimModule_basic",
|
||||||
|
"IndexSelectRank0IdxModule_basic",
|
||||||
|
"IndexSelectSingleIdxModule_basic",
|
||||||
|
"IndexSelectTwoIdxModule_basic",
|
||||||
|
"IndexSelectWholeDimensionModule_basic",
|
||||||
|
"IndexSelectWholeTensorModule_basic",
|
||||||
|
"IndexTensorDyanmicInputContiguousWithNoneModule_basic",
|
||||||
|
"IndexTensorDyanmicInputNonContiguousWithNoneModule_basic",
|
||||||
|
"IndexTensorHackedTwinMultiInputNonContiguousMultipleStaticDims_basic",
|
||||||
|
"IndexTensorMultiInputContiguousCenter_basic",
|
||||||
|
"IndexTensorMultiInputContiguousOneDimDynamic_basic",
|
||||||
|
"IndexTensorMultiInputNonContiguousDynamic_basic",
|
||||||
|
"IndexTensorMultiInputNonContiguousMultipleStaticDims_basic",
|
||||||
|
"IndexTensorMultiInputNonContiguousOneDimDynamic_basic",
|
||||||
|
"IndexTensorMultiInputNonContiguous_basic",
|
||||||
|
"IndexTensorMultiInputOneDim_basic",
|
||||||
|
"IndexTensorMultiInputThreeIndexers_basic",
|
||||||
|
"IndexTensorMultiInput_basic",
|
||||||
|
"IndexTensorNegativeIndexModule_basic",
|
||||||
|
"IndexTensorSelectDimModule_basic",
|
||||||
|
"IndexTensorStaticContiguousWithNoneModule_basic",
|
||||||
|
"IndexTensorStaticNonContiguousWithNoneModule_basic",
|
||||||
|
"IntFloatModule_basic",
|
||||||
|
"IntImplicitModule_basic",
|
||||||
|
"IsFloatingPointFloat_True",
|
||||||
|
"IsFloatingPointInt_False",
|
||||||
|
"LayerNormLastDimModule_basic",
|
||||||
|
"LayerNormModule_basic",
|
||||||
|
"LayerNormNormalizeOverAllDimsModule_basic",
|
||||||
|
"LenStrModule_basic",
|
||||||
|
"LinalgNormKeepDimComplexModule_basic",
|
||||||
|
"LinalgVectorNormComplexModule_basic",
|
||||||
|
"LinspaceDtypeModule_basic",
|
||||||
|
"LinspaceEmptyModule_basic",
|
||||||
|
"LinspaceModule_basic",
|
||||||
|
"LinspaceOneSizeModule_basic",
|
||||||
|
"LinspaceTwoSizeModule_basic",
|
||||||
|
"LogSoftmaxIntModule_basic",
|
||||||
|
"MaskedFillTensorFloatValueModule_basic",
|
||||||
|
"MatmulBroadcastBatchDim_basic",
|
||||||
|
"MatmulStaticBroadcast_basic",
|
||||||
|
"MaxPool1dCeilModeTrueModule_basic",
|
||||||
|
"MaxPool1dModule_basic",
|
||||||
|
"MaxPool2dCeilModeTrueModule_basic",
|
||||||
|
"MaxPool2dModule_basic",
|
||||||
|
"MaxPool2dWithIndicesAllNegativeValuesModule_basic",
|
||||||
|
"MaxPool2dWithIndicesAllOnesModule_basic",
|
||||||
|
"MaxPool2dWithIndicesBackwardDynamic3DModule_basic",
|
||||||
|
"MaxPool2dWithIndicesBackwardDynamic4DModule_basic",
|
||||||
|
"MaxPool2dWithIndicesBackwardStatic3DModule_basic",
|
||||||
|
"MaxPool2dWithIndicesBackwardStatic4DModule_basic",
|
||||||
|
"MaxPool2dWithIndicesCeilModeTrueModule_basic",
|
||||||
|
"MaxPool2dWithIndicesFullSizeKernelModule_basic",
|
||||||
|
"MaxPool2dWithIndicesModule_basic",
|
||||||
|
"MaxPool2dWithIndicesNonDefaultDilationModule_basic",
|
||||||
|
"MaxPool2dWithIndicesNonDefaultPaddingModule_basic",
|
||||||
|
"MaxPool2dWithIndicesNonDefaultParamsModule_basic",
|
||||||
|
"MaxPool2dWithIndicesNonDefaultStrideModule_basic",
|
||||||
|
"MaxPool2dWithIndicesStaticModule_basic",
|
||||||
|
"MaxPool3dCeilModeTrueModule_basic",
|
||||||
|
"MaxPool3dEmptyStrideStaticModule_basic",
|
||||||
|
"MaxPool3dLargeDatadModule_basic",
|
||||||
|
"MaxPool3dModuleRandomSimple_basic",
|
||||||
|
"MaxPool3dModule_basic",
|
||||||
|
"MaxPool3dStaticCeilModeTrueModule_basic",
|
||||||
|
"MaxPool3dStaticModule_basic",
|
||||||
|
"MeanDimDtypeModule_basic",
|
||||||
|
"MeanDimEmptyDimModule_basic",
|
||||||
|
"MeanDimNoneDimModule_basic",
|
||||||
|
"MeanDtypeModule_basic",
|
||||||
|
"MseLossMeanReductionModule_basic",
|
||||||
|
"MseLossSumReductionWithDifferentElemTypeModule_basic",
|
||||||
|
"MulFloatModule_basic",
|
||||||
|
"MulIntModule_basic",
|
||||||
|
"NativeBatchNorm1DModule_basic",
|
||||||
|
"NativeBatchNorm2DModule_basic",
|
||||||
|
"NativeBatchNorm3DModule_basic",
|
||||||
|
"NativeBatchNormNoneWeightModule_basic",
|
||||||
|
"NativeDropoutTrainModule_basic",
|
||||||
|
"NativeDropoutTrainStaticShapeModule_basic",
|
||||||
|
"NativeGroupNormBackwardModule_basic",
|
||||||
|
"NativeGroupNormModule_basic",
|
||||||
|
"NativeLayerNormDynamicModule_basic",
|
||||||
|
"NativeLayerNormModule4D_basic",
|
||||||
|
"NativeLayerNormModule_basic",
|
||||||
|
"NeFloatIntModule_basic",
|
||||||
|
"NeIntModule_basic",
|
||||||
|
"NewEmptyModuleDefaultDtype_basic",
|
||||||
|
"NewEmptyModuleFalsePinMemory_basic",
|
||||||
|
"NewEmptyModuleFloat2D_basic",
|
||||||
|
"NewEmptyModuleFloat3D_basic",
|
||||||
|
"NewEmptyModuleInt2D_basic",
|
||||||
|
"NewEmptyModuleInt3D_basic",
|
||||||
|
"NewEmptyModuleLayoutIntDtype_basic",
|
||||||
|
"NewEmptyModuleNonDefaultFloatDtype_basic",
|
||||||
|
"NewEmptyModuleNonDefaultIntDtype_basic",
|
||||||
|
"NewEmptyStridedModuleDefaultDtype_basic",
|
||||||
|
"NewFullModuleInt2D_basic",
|
||||||
|
"NewFullModuleInt3D_basic",
|
||||||
|
"NllLossModuleBackward1DMeanWeight_basic",
|
||||||
|
"NllLossModuleBackward1DMean_basic",
|
||||||
|
"NllLossModuleBackward1DSumWeight_basic",
|
||||||
|
"NllLossModuleBackward1DSum_basic",
|
||||||
|
"NllLossModuleBackward1DWeight_basic",
|
||||||
|
"NllLossModuleBackward1D_basic",
|
||||||
|
"NllLossModuleBackwardMeanWeight_basic",
|
||||||
|
"NllLossModuleBackwardMean_basic",
|
||||||
|
"NllLossModuleBackwardSumWeight_basic",
|
||||||
|
"NllLossModuleBackwardSum_basic",
|
||||||
|
"NllLossModuleBackwardWeight_basic",
|
||||||
|
"NllLossModuleBackward_basic",
|
||||||
|
"NllLossModuleBackward_ignore_index",
|
||||||
|
"NllLossModule_1D_basic",
|
||||||
|
"NllLossModule_mean_basic",
|
||||||
|
"NllLossModule_sum_basic",
|
||||||
|
"NormScalarComplexModule_basic",
|
||||||
|
"NormScalarModule_basic",
|
||||||
|
"NormScalarOptDimKeepDimComplexModule_basic",
|
||||||
|
"NormalFunctionalModule_basic",
|
||||||
|
"NumToTensorFloatModule_basic",
|
||||||
|
"NumToTensorIntModule_basic",
|
||||||
|
"NumelModule_basic",
|
||||||
|
"NumelZeroRankModule_basic",
|
||||||
|
"OnesLikeModule_falsePinMemory",
|
||||||
|
"PixelShuffleModuleFullDynamic_basic",
|
||||||
|
"PixelShuffleModuleSpatiallyDynamic_basic",
|
||||||
|
"PixelShuffleModuleSpatiallyStatic_basic",
|
||||||
|
"PixelShuffleModuleStaticRank3Int64_basic",
|
||||||
|
"PixelShuffleModuleStaticRank4Float32_basic",
|
||||||
|
"PowIntFloatModule_basic",
|
||||||
|
"PrimMaxIntModule_basic",
|
||||||
|
"PrimMinIntDynamicModule_basic",
|
||||||
|
"PrimMinIntModule_basic",
|
||||||
|
"PrimsConvertElementTypeModule_basic",
|
||||||
|
"PrimsSqueezeEmptyDimensionsModule_basic",
|
||||||
|
"PrimsSqueezeModule_basic",
|
||||||
|
"PrimsViewOfModule_basic",
|
||||||
|
"PrimsViewOfZeroRankModule_basic",
|
||||||
|
"QuantizedBatchedInputSingleLayer_basic",
|
||||||
|
"QuantizedMLP_basic",
|
||||||
|
"QuantizedNoLayer_basic",
|
||||||
|
"QuantizedReluInt32_basic",
|
||||||
|
"QuantizedReluInt8_basic",
|
||||||
|
"QuantizedReluUint8_basic",
|
||||||
|
"QuantizedSingleLayer_basic",
|
||||||
|
"RandIntDtypeModule_basic",
|
||||||
|
"RandIntLowDtypeModule_basic",
|
||||||
|
"RandIntLowModule_basic",
|
||||||
|
"RandIntModule_basic",
|
||||||
|
"RandIntPinMemoryModule_basic",
|
||||||
|
"RandLikeDtypeModule_basic",
|
||||||
|
"RandLikeModule_basic",
|
||||||
|
"RandModule_basic",
|
||||||
|
"RandnDtypeDeviceModule_basic",
|
||||||
|
"RandnGeneratorF64Module_basic",
|
||||||
|
"RandnGeneratorModule_basic",
|
||||||
|
"RandnLikeDtypeModule_basic",
|
||||||
|
"RandnLikeModule_basic",
|
||||||
|
"RandnModule_basic",
|
||||||
|
"ReduceAllDimBool_basic",
|
||||||
|
"ReduceAllDimEmpty_basic",
|
||||||
|
"ReduceAllDimFloat_basic",
|
||||||
|
"ReduceAllDimInt_basic",
|
||||||
|
"ReduceAllFloatModule_basic",
|
||||||
|
"ReduceAllIntModule_basic",
|
||||||
|
"ReduceAnyFloatModule_basic",
|
||||||
|
"ReduceAnyIntModule_basic",
|
||||||
|
"ReduceFrobeniusNormComplexModule_basic",
|
||||||
|
"ReduceL1NormComplexModule_basic",
|
||||||
|
"ReduceL1NormWithDTypeModule_basic",
|
||||||
|
"ReduceL2NormComplexModule_basic",
|
||||||
|
"ReduceL3NormAllDimsModule_basic",
|
||||||
|
"ReduceL3NormKeepDimComplexModule_basic",
|
||||||
|
"ReduceL3NormKeepDimModule_basic",
|
||||||
|
"ReduceMaxAllDims_basic",
|
||||||
|
"ReduceMaxAlongDimNegative_basic",
|
||||||
|
"ReduceMaxAlongDimUnsignedInt_basic",
|
||||||
|
"ReduceMaxAlongDim_basic",
|
||||||
|
"ReduceMaxFloatModule_basic",
|
||||||
|
"ReduceMaxKeepDim_basic",
|
||||||
|
"ReduceMaxSignedIntModule_basic",
|
||||||
|
"ReduceMaxUnsignedIntModule_basic",
|
||||||
|
"ReduceMinAlongDimNegative_basic",
|
||||||
|
"ReduceMinAlongDimSignedInt_basic",
|
||||||
|
"ReduceMinAlongDimUnsignedInt_basic",
|
||||||
|
"ReduceMinAlongDim_basic",
|
||||||
|
"ReduceMinFloatModule_basic",
|
||||||
|
"ReduceMinKeepDimReturnBoth_basic",
|
||||||
|
"ReduceMinKeepDim_basic",
|
||||||
|
"ReduceMinSignedIntModule_basic",
|
||||||
|
"ReduceMinUnsignedIntModule_basic",
|
||||||
|
"ReduceProdDimIntFloatModule_basic",
|
||||||
|
"ReduceProdDtypeFloatModule_basic",
|
||||||
|
"ReduceProdDtypeIntModule_basic",
|
||||||
|
"ReduceProdElementTypeBoolModule_basic",
|
||||||
|
"ReduceProdFloatModule_basic",
|
||||||
|
"ReduceProdSignedIntModule_basic",
|
||||||
|
"ReduceProdUnsignedIntModule_basic",
|
||||||
|
"ReduceSumDimIntListDtypeFloatModule_basic",
|
||||||
|
"ReduceSumDimIntListDtypeIntModule_basic",
|
||||||
|
"ReduceSumDimIntListElementTypeBoolModule_basic",
|
||||||
|
"ReduceSumDimIntListEmptyDimModule_basic",
|
||||||
|
"ReduceSumDtypeFloatModule_basic",
|
||||||
|
"ReduceSumDtypeIntModule_basic",
|
||||||
|
"ReduceSumElementTypeBoolModule_basic",
|
||||||
|
"ReflectionPad1dModule2dInput_Right",
|
||||||
|
"ReflectionPad1dModule2dInput_basic",
|
||||||
|
"ReflectionPad1dModule3dInput_Left",
|
||||||
|
"ReflectionPad1dModule3dInput_basic",
|
||||||
|
"ReflectionPad2dModule_Bottom",
|
||||||
|
"ReflectionPad2dModule_Left",
|
||||||
|
"ReflectionPad2dModule_Right",
|
||||||
|
"ReflectionPad2dModule_Top",
|
||||||
|
"ReflectionPad2dModule_basic",
|
||||||
|
"ReplicationPad2dModule_basic",
|
||||||
|
"ReplicationPad2dModule_bottom0",
|
||||||
|
"ReplicationPad2dModule_left0",
|
||||||
|
"ReplicationPad2dModule_right0",
|
||||||
|
"ReplicationPad2dModule_top0",
|
||||||
|
"RollModule_basic",
|
||||||
|
"RsubInt0d_NumToTensor_Module_basic",
|
||||||
|
"RsubIntModule_basic",
|
||||||
|
"RsubIntModule_noalpha_basic",
|
||||||
|
"ScalarConstantTupleModule_basic",
|
||||||
|
"ScalarImplicitFloatModule_basic",
|
||||||
|
"ScalarImplicitIntModule_basic",
|
||||||
|
"ScaledDotProductAttentionDifferentModule_basic",
|
||||||
|
"ScatterReduceFloatMaxModule",
|
||||||
|
"ScatterReduceFloatMaxModuleIncludeSelf",
|
||||||
|
"ScatterReduceFloatMeanModule",
|
||||||
|
"ScatterReduceFloatMeanModuleIncludeSelf",
|
||||||
|
"ScatterReduceFloatMinModule",
|
||||||
|
"ScatterReduceFloatMinModuleIncludeSelf",
|
||||||
|
"ScatterReduceFloatProdModule",
|
||||||
|
"ScatterReduceFloatProdModuleIncludeSelf",
|
||||||
|
"ScatterReduceFloatSumModule",
|
||||||
|
"ScatterReduceFloatSumModuleIncludeSelf",
|
||||||
|
"ScatterReduceIntMaxModule",
|
||||||
|
"ScatterReduceIntMaxModuleIncludeSelf",
|
||||||
|
"ScatterReduceIntMeanModule",
|
||||||
|
"ScatterReduceIntMeanModuleIncludeSelf",
|
||||||
|
"ScatterReduceIntMinModule",
|
||||||
|
"ScatterReduceIntMinModuleIncludeSelf",
|
||||||
|
"ScatterReduceIntProdModule",
|
||||||
|
"ScatterReduceIntProdModuleIncludeSelf",
|
||||||
|
"ScatterReduceIntSumModule",
|
||||||
|
"ScatterReduceIntSumModuleIncludeSelf",
|
||||||
|
"ScatterSrcModule_basic",
|
||||||
|
"ScatterSrcStaticModule_basic",
|
||||||
|
"ScatterValueFloatModule_basic",
|
||||||
|
"ScatterValueIntModule_basic",
|
||||||
|
"SelectScattertModule_basic",
|
||||||
|
"SelectScattertStaticModule_basic",
|
||||||
|
"SliceCopyEndGreaterThanDimSize_Module_basic",
|
||||||
|
"SliceCopyNegative_Module_basic",
|
||||||
|
"SliceCopyNonZeroDim_Module_basic",
|
||||||
|
"SliceCopyStartGreaterThanDimSize_Module_basic",
|
||||||
|
"SliceCopy_Module_basic",
|
||||||
|
"SliceEndSleStartModule_basic",
|
||||||
|
"SliceOutOfLowerBoundEndIndexModule_basic",
|
||||||
|
"SliceOutOfLowerBoundStartIndexModule_basic",
|
||||||
|
"SliceScatterModule_basic",
|
||||||
|
"SliceScatterNegativeDimModule_basic",
|
||||||
|
"SliceScatterNegativeEndModule_basic",
|
||||||
|
"SliceScatterStaticModule_basic",
|
||||||
|
"SliceScatterStepVariationModule_basic",
|
||||||
|
"SliceScatterZeroDimModule_basic",
|
||||||
|
"SliceSizeTwoStepModule_basic",
|
||||||
|
"SoftmaxIntArgTypeF64Module_basic",
|
||||||
|
"SoftmaxIntNonNoneDtypeModule_basic",
|
||||||
|
"SoftplusModule_basic",
|
||||||
|
"SortIntListReverse_basic",
|
||||||
|
"SortIntList_basic",
|
||||||
|
"SortTensorDescending_basic",
|
||||||
|
"SortTensorInteger_basic",
|
||||||
|
"SortTensorNegativeDimension_basic",
|
||||||
|
"SortTensorSpecificDimension_basic",
|
||||||
|
"SortTensor_basic",
|
||||||
|
"SplitDimDynamicModule_basic",
|
||||||
|
"SplitDimStaticModule_basic",
|
||||||
|
"SqrtIntConstantModule_basic",
|
||||||
|
"SqrtIntModule_basic",
|
||||||
|
"StdBiasedModule_basic",
|
||||||
|
"StdCorrectionAllDimReduceModule_basic",
|
||||||
|
"StdCorrectionEmptyDimModule_basic",
|
||||||
|
"StdCorrectionKeepDimModule_basic",
|
||||||
|
"StdCorrectionLargeInputModule_basic",
|
||||||
|
"StdCorrectionModule_basic",
|
||||||
|
"StdCorrectionNoneModule_basic",
|
||||||
|
"StdCorrectionSingleDimReduceModule_basic",
|
||||||
|
"StdDimBiasedModule_basic",
|
||||||
|
"StdDimEmptyDimModule_basic",
|
||||||
|
"StdDimKeepDimFalseModule_basic",
|
||||||
|
"StdDimKeepDimTrueModule_basic",
|
||||||
|
"StdDimNoneDimModule_basic",
|
||||||
|
"StdUnbiasedModule_basic",
|
||||||
|
"SubFloatModule_basic",
|
||||||
|
"SubIntModule_basic",
|
||||||
|
"TModuleRank0_basic",
|
||||||
|
"TensorToBoolZeroRank_basic",
|
||||||
|
"TensorToBool_basic",
|
||||||
|
"TensorToFloatZeroRank_basic",
|
||||||
|
"TensorToFloat_basic",
|
||||||
|
"TensorToIntZeroRank_basic",
|
||||||
|
"TensorToInt_basic",
|
||||||
|
"TensorsConcatPromoteDTypeModule_basic",
|
||||||
|
"TensorsStackPromoteDTypeModule_basic",
|
||||||
|
"TestMultipleTensorAndPrimitiveTypesReturn_basic",
|
||||||
|
"Threshold1dIntModule_basic",
|
||||||
|
"Threshold2dIntModule_basic",
|
||||||
|
"Threshold3dIntModule_basic",
|
||||||
|
"ThresholdBackward1dFloatModule_basic",
|
||||||
|
"ThresholdBackward1dIntModule_basic",
|
||||||
|
"ThresholdBackward1dMixedModule_basic",
|
||||||
|
"ThresholdBackward2dFloatModule_basic",
|
||||||
|
"ThresholdBackward2dIntModule_basic",
|
||||||
|
"ThresholdBackward2dMixedModule_basic",
|
||||||
|
"ThresholdBackward3dFloatModule_basic",
|
||||||
|
"ThresholdBackward3dIntModule_basic",
|
||||||
|
"ThresholdBackward3dMixedModule_basic",
|
||||||
|
"ToCopyWithDTypeFalsePinMemoryModule_basic",
|
||||||
|
"ToCopyWithDTypeModule_basic",
|
||||||
|
"TorchPrimLoopForLikeModule_basic",
|
||||||
|
"TorchPrimLoopWhileLikeModule_basic",
|
||||||
|
"TraceModule_basic",
|
||||||
|
"TraceModule_empty",
|
||||||
|
"TraceModule_nonsquare",
|
||||||
|
"TraceSignedIntModule_basic",
|
||||||
|
"TraceUnsignedIntModule_basic",
|
||||||
|
"TraceUnsignedIntModule_empty",
|
||||||
|
"TypeConversionI1ToF64Module_basic",
|
||||||
|
"TypeConversionI1ToI32Module_basic",
|
||||||
|
"UnbindIntGetItem_Module_basic",
|
||||||
|
"UnbindIntListUnpack_Module_basic",
|
||||||
|
"UniformModule_basic",
|
||||||
|
"UniformNoCorrelationModule_basic",
|
||||||
|
"UniformStaticShapeModule_basic",
|
||||||
|
"UnsafeIndexPutHackedTwin1DFloatNonAccumulateModule_basic",
|
||||||
|
"UnsafeViewCollapseDynamicWithAtenSizeIntModule_basic",
|
||||||
|
"UpSampleNearest2dBackwardScalesNone_basic",
|
||||||
|
"UpSampleNearest2dBackward_basic",
|
||||||
|
"UpSampleNearest2dDynamicFactor_basic",
|
||||||
|
"UpSampleNearest2dDynamicSize_basic",
|
||||||
|
"UpSampleNearest2dStaticFactor_basic",
|
||||||
|
"UpSampleNearest2dStaticSize_basic",
|
||||||
|
"UpSampleNearest2d_basic",
|
||||||
|
"VarBiasedModule_basic",
|
||||||
|
"VarCorrectionAllDimReduceModule_basic",
|
||||||
|
"VarCorrectionEmptyDimModule_basic",
|
||||||
|
"VarCorrectionKeepDimModule_basic",
|
||||||
|
"VarCorrectionLargeInputModule_basic",
|
||||||
|
"VarCorrectionModule_basic",
|
||||||
|
"VarCorrectionNoneModule_basic",
|
||||||
|
"VarCorrectionSingleDimReduceModule_basic",
|
||||||
|
"VarDimAllDimReduceModule_basic",
|
||||||
|
"VarDimBiasedModule_basic",
|
||||||
|
"VarDimEmptyDimModule_basic",
|
||||||
|
"VarDimModule_basic",
|
||||||
|
"VarDimMultiDimModule_basic",
|
||||||
|
"VarDimNegativeModule_basic",
|
||||||
|
"VarDimNoneDimModule_basic",
|
||||||
|
"VarDimSingleDimModule_basic",
|
||||||
|
"VarDimUnbiasedModule_basic",
|
||||||
|
"VarMeanBiasedModule_basic",
|
||||||
|
"VarMeanCorrectionModule_basic",
|
||||||
|
"VarMeanCorrectionNoneModule_basic",
|
||||||
|
"VarMeanDimBiasedModule_basic",
|
||||||
|
"VarMeanDimModule_basic",
|
||||||
|
"VarMeanUnbiasedModule_basic",
|
||||||
|
"VarUnbiasedModule_basic",
|
||||||
|
"ViewCollapseDynamicWithAtenSizeIntModule_basic",
|
||||||
|
"ViewSizeFromOtherTensor_basic",
|
||||||
|
"ZeroFloat32Module_basic",
|
||||||
|
"ZeroInt32Module_basic",
|
||||||
|
"ZeroInt64Module_basic",
|
||||||
|
"ZerosLikeModule_falsePinMemory",
|
||||||
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue