[PIR]Fix infershape error when infershape function has MetaConfig (#62010)

0x45f · web-flow · commit a68d9ccbd684 · 2024-02-23T19:51:13.000+08:00
diff --git a/paddle/fluid/pir/dialect/op_generator/op_build_gen.py b/paddle/fluid/pir/dialect/op_generator/op_build_gen.py
@@ -14,20 +14,110 @@
 
 # generator build function
 _INFERMETA_NEED_META_CONFIG = {
-    'SplitInferMeta',
-    'SumInferMeta',
-    'SplitWithNumInferMeta',
+    # binary.h
+    'AllValueCompareInferMeta',
+    'KLDivInferMeta',
+    'ArrayWriteInferMeta',
+    'ArrayReadInferMeta',
+    'BCELossInferMeta',
+    'BinomialInferMeta',
+    'ConvInferMeta',
+    'Conv3DInferMeta',
+    'ConvTransposeInferMeta',
+    'Conv2dTransposeInferMeta',
+    'CrossEntropyWithSoftmaxInferMeta',
+    'CSoftmaxWithCrossEntropyInferMeta',
+    'DepthwiseConvInferMeta',
+    'DistributeFpnProposalsInferMeta',
+    'ElementwiseRawInferMeta',
+    'GridSampleBaseInferMeta',
+    'HuberLossInferMeta',
+    'IndexSampleInferMeta',
+    'LogLossInferMeta',
+    'MarginCrossEntropyInferMeta',
+    'MatrixNMSInferMeta',
+    'PReluInferMeta',
+    'SegmentPoolInferMeta',
+    'YoloBoxInferMeta',
+    'ValueCompareInferMeta',
+    'UnpoolInferMeta',
+    'Unpool3dInferMeta',
+    # fusion.h
+    'FusedAttentionInferMeta',
+    'BNActXPUInferMeta',
+    'FusedFCElementwiseLayerNormInferMeta',
+    'FusedConv2dAddActInferMeta',
+    # multiary.h
+    'AddNTensorArrayInferMeta',
+    'AddNInferMeta',
+    'AucInferMeta',
+    'BatchNormInferMeta',
+    'BatchNormInferInferMeta',
+    'BilinearInferMeta',
+    'CoalesceTensorInferMeta',
+    'CheckMemoryContinueInferMeta',
     'ConcatInferMeta',
+    'DeformableConvInferMeta',
+    'FusedBiasActInferMeta',
+    'InterpolateInferMeta',
+    'NceInferMeta',
+    'SigmoidCrossEntropyWithLogitsInferMeta',
+    'StackInferMeta',
+    'FusedConvInferMeta',
+    # nullary.h
+    'EyeInferMeta',
+    # ternary.h
+    'AccuracyInferMeta',
+    'BoxCoderInferMeta',
+    'InstanceNormInferMeta',
+    'LayerNormInferMeta',
+    'MatchMatrixTensorInferMeta',
+    'MultiClassNMSInferMeta',
+    'NllLossRawInferMeta',
+    'RoiAlignInferMeta',
+    'SpectralNormInferMeta',
+    'ViterbiDecodeInferMeta',
+    'TdmSamplerInferMeta',
+    # unary.h
+    'GetSplitAxisValue',
+    'ArgMinMaxInferMeta',
+    'ArrayToTensorInferMeta',
+    'CropInferMeta',
+    'EigvalsInferMeta',
+    'FractionalMaxPoolInferMeta',
+    'MaxPoolWithIndexInferMeta',
+    'MaxPoolV2InferMeta',
+    'MultinomialInferMeta',
+    'OverlapAddInferMeta',
+    'PadInferMeta',
+    'Pad3dInferMeta',
+    'PoolInferMeta',
+    'Pool2DInferMeta',
+    'ReduceIntArrayAxisInferMetaBase',
     'ReduceIntArrayAxisInferMeta',
+    'ReshapeInferMeta',
     'ReshapeWithXShapeInferMeta',
+    'ReverseInferMeta',
+    'ReverseArrayInferMeta',
+    'ShardIndexInferMeta',
+    'SliceArrayInferMeta',
+    'SliceArrayDenseInferMeta',
     'SliceRawInferMeta',
-    'StackInferMeta',
-    'Conv2dTransposeInferMeta',
-    'FusedConv2dAddActInferMeta',
-    'InterpolateInferMeta',
-    'DeformableConvInferMeta',
-    'MatrixNMSInferMeta',
-    'IndexSampleInferMeta',
+    'SplitInferMeta',
+    'SplitWithNumInferMeta',
+    'SqueezeInferMeta',
+    'SqueezeWithXShapeInferMeta',
+    'StridedSliceRawInferMeta',
+    'StridedSliceInferMeta',
+    'SumInferMeta',
+    'SumRawInferMeta',
+    'TemporalShiftInferMeta',
+    'TileInferMeta',
+    'TopKInferMeta',
+    'UnfoldInferMeta',
+    'UnsqueezeInferMeta',
+    'UnsqueezeWithXShapeInferMeta',
+    'ArrayPopInferMeta',
 }
 
 _PREPARE_DATA_WITH_VECTOR_INT64_MTTABLE_ATTRIBUTE = {'FrobeniusNormOp'}
diff --git a/paddle/phi/infermeta/unary.cc b/paddle/phi/infermeta/unary.cc
@@ -3595,6 +3595,7 @@ void ReshapeInferMeta(const MetaTensor& x,
   if (!config.is_runtime && shape.FromTensor()) {
     out->set_dims(common::make_ddim(shape_data));
     out->share_lod(x);
+    out->set_dtype(x.dtype());
     return;
   }
   InferMetaFromVecValue(x, shape_data, out);

Original file line number	Diff line number	Diff line change
`@@ -3595,6 +3595,7 @@ void ReshapeInferMeta(const MetaTensor& x,`
`3595`	`3595`	`if (!config.is_runtime && shape.FromTensor()) {`
`3596`	`3596`	`out->set_dims(common::make_ddim(shape_data));`
`3597`	`3597`	`out->share_lod(x);`
	`3598`	`+ out->set_dtype(x.dtype());`
`3598`	`3599`	`return;`
`3599`	`3600`	`}`
`3600`	`3601`	`InferMetaFromVecValue(x, shape_data, out);`