diff --git a/torch_npu/csrc/framework/aoe/AoeUtils.h b/torch_npu/csrc/framework/aoe/AoeUtils.h index 9643d15adbf9e79730257ad99d5199b6d4411a1d..fd28de629208a5ae8acc1a26f2b8ccad41318ae2 100644 --- a/torch_npu/csrc/framework/aoe/AoeUtils.h +++ b/torch_npu/csrc/framework/aoe/AoeUtils.h @@ -43,112 +43,112 @@ public: aclGraphDumpOption* AclGraphDumpOption=NULL; std::unordered_set whilte_list_ = { "Abs", - "AbsGrad", - "AcosGrad", + "AccumulateNV2", "Add", - "AsinGrad", - "AsinhGrad", - "AtanGrad", - "AvgPool", - "BatchMatMul", - "BatchMatMulV2", + "AddN", + "Adds", + "BatchNorm3D", + "BiasAdd", "BiasAddGrad", + "BNInfer", + "BNTrainingReduce", "BNTrainingUpdate", - "Cast", - "Ceil", - "ConcatD", - "Conv2D", - "Conv2DBackpropFilter", - "Conv2DBackpropInput", - "Conv2DCompress", - "Conv2DTranspose", - "Conv3D", - "Conv3DBackpropFilter", - "Conv3DBackpropInput", - "Cos", - "Cosh", - "CosineEmbeddingLoss", - "Deconvolution", - "DepthwiseConv2D", - "DepthwiseConv2DBackpropFilter", - "DepthwiseConv2DBackpropInput", + "BNTrainingUpdateGrad", + "BNTrainingUpdateV2", + "BNTrainingUpdateV3", + "BroadcastToD", + "ClipByNormNoDivSum", + "ClipByValue", + "DiagPartD", "Div", + "DivNoNan", "DynamicRNN", "Elu", "EluGrad", "Equal", - "Erf", - "Erfc", "Exp", - "Expm1", + "ExpandD", "Floor", - "FullyConnection", - "FullyConnectionCompress", + "FloorDiv", "Gelu", - "GeluGrad", - "GEMM", - "GNTrainingReduce", - "GNTrainingUpdate", - "INTrainingReduceV2", - "INTrainingUpdateV2", - "Inv", - "InvGrad", + "Greater", + "GreaterEqual", + "InstanceNorm", "L2Loss", - "L2Normalize", - "L2NormalizeGrad", + "LambUpdateWithLrV2", + "LayerNormBetaGammaBackpropV2", + "LeakyRelu", + "Less", + "LessEqual", "Log", "Log1p", + "LogicalAnd", + "LogicalNot", + "LogicalOr", "LogSoftmaxGrad", "LogSoftmaxV2", - "MatMul", - "MatMulV2", - "MatMulV2Compress", + "LpNorm", + "MatrixDiagD", "Maximum", - "Mod", + "Minimum", "Mul", + "Muls", "Neg", - "OnesLike", - "Pooling", + "NotEqual", "Pow", - "PReluGrad", - "Reciprocal", - "ReciprocalGrad", + "PRelu", + "RealDiv", "ReduceAllD", - "ReduceAnyD", "ReduceMaxD", "ReduceMeanD", + "ReduceMinD", "ReduceSumD", "Relu", - "Relu6", - "Relu6Grad", - "ReluGrad", - "Rint", + "ReluV2", "Round", "Rsqrt", "RsqrtGrad", - "Selu", "Sigmoid", - "SigmoidCrossEntropyWithLogits", - "SigmoidGrad", "Sign", - "Sinh", - "SmoothL1Loss", "SoftmaxCrossEntropyWithLogits", - "SoftmaxGrad", "SoftmaxV2", "Softplus", - "Softsign", - "SplitD", - "Sqrt", + "SplitVD", "SqrtGrad", "Square", + "SquaredDifference", + "SquareSumV1", + "SquareSumV2", "StridedSliceD", "Sub", "Tanh", - "TanhGrad", "TileD", - "Unpack"}; - + "Unpack", + "AvgPool", + "Deconvolution", + "Conv2D", + "Conv2DCompress", + "Conv2DBackpropInput", + "Conv2DBackpropFilter", + "GEMM", + "MatMul", + "MatMulV2", + "MatMulV2Compress", + "BatchMatMul", + "BatchMatMulV2", + "FullyConnection", + "FullyConnectionCompress", + "DepthwiseConv2D", + "DepthwiseConv2DBackpropInput", + "DepthwiseConv2DBackpropFilter", + "Conv3D", + "Conv3DBackpropInput", + "Conv3DBackpropFilter", + "AvgPool3DGrad", + "Pooling", + "Conv2DTranspose", + "Conv3DTranspose"}; + }; AoeDumpGraphManager& aoe_manager();