[PTen] Unify InferMeta(Shape) Function in pten and fluid op

[chenwhql]

PR types

Function optimization

PR changes

Others

Describe

[PTen] Upgrade InferMeta and ArgumentMapping design

一、解决之前pten迁移InferMeta函数没有和原先Op的InferShape统一维护一份实现的问题

1. 设计MetaTensor接口，支持必要的Meta成员读写方法，在pten中，MetaTensor的成员为TensorBase*指针，以兼容DenseTensor，SelectedRows，SparseTensor等多种tensor类型，以及原通过继承兼容原fluid的VarDesc，Variable

以sign为例，sign的InferMeta方法为UnchangedInferMeta，升级前后的写法分别为

// 升级前 DenseTensorMeta UnchangedInferMeta(const DenseTensorMeta& x_meta) { return x_meta; } // 升级后，MetaConfig参数在后续会移到末尾 void UnchangedInferMetaNew(MetaConfig config, const MetaTensor& x, MetaTensor* out) { out->set_dims(x.dims()); out->share_lod(x); } 

此处的MetaConfig，用来存储一些InferShape需要的状态变量，类似Context，但是相比Context来讲，非常轻量，在内部函数复用时一般也用不到（可能放到末尾会更自然）

struct MetaConfig { bool is_runtime{true}; // 用于判断当前处于编译器还是执行期，统一两种场景的infershape需求 MetaConfig() = default; MetaConfig(bool is_runtime) : is_runtime(is_runtime) {} }; 

2. 实现类似Kernel形式归一化的PT_KERNEL宏，这里命名为PT_INFER_META

然后将PT_INFER_META(UnchangedInferMetaNew)包装到一个functor中，functor中先将InferShapeContext转换为InferMetaContext，再调用相应InferMeta函数，可以通过一个宏整理代码

然后将该functor在Op注册时维护到相应OpInfo中即可，同时删除原先Op的InferShape实现，示例如下

// 原先实现 class SignOp : public framework::OperatorWithKernel { public: using framework::OperatorWithKernel::OperatorWithKernel; void InferShape(framework::InferShapeContext *ctx) const override { OP_INOUT_CHECK(ctx->HasInput("X"), "Input", "X", "sign"); OP_INOUT_CHECK(ctx->HasOutput("Out"), "Output", "Out", "sign"); ctx->SetOutputDim("Out", ctx->GetInputDim("X")); ctx->ShareLoD("X", /*->*/ "Out"); } }; namespace ops = paddle::operators; REGISTER_OPERATOR(sign, ops::SignOp, ops::SignOpMaker<float>, ops::SignGradMaker<paddle::framework::OpDesc>, ops::SignGradMaker<paddle::imperative::OpBase>); // 升级后实现 class SignOp : public framework::OperatorWithKernel { public: using framework::OperatorWithKernel::OperatorWithKernel; }; DELCARE_INFER_SHAPE_FUNCTOR( sign, SignInferShapeFunctor, PT_INFER_META(pten::UnchangedInferMetaNew)); REGISTER_OPERATOR(sign, ops::SignOp, ops::SignOpMaker<float>, ops::SignGradMaker<paddle::framework::OpDesc>, ops::SignGradMaker<paddle::imperative::OpBase>, SignInferShapeFunctor); 

像这样，原Op的InferShape函数迁移至pten InferMeta之后，可以重新注册回fluid中被调用，从而实现InferShape的函数化复用与全局统一。

TODO：目前只改写了sign验证通过，其他kernel的改写在后续PR逐步进行

二、新增OpUtilsMap，用于注册Op对应ApiName和ArgumentMappingFn，以兼容原fluid以及infrt

ArgumentMappingFn注册，通过在cc文件中注册对应函数，编译时自动生成对应编译对象，实现注册

// op_type, arg_mapping_fn PT_REGISTER_ARG_MAPPING_FN(scale, pten::ScaleOpArgumentMapping); 

TODO：其他kernel的改写在后续PR进行
TODO：InferMeta的注册在后续PR实现

[paddle-bot-old]

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

[zhiqiu]

LGTM

[zhiqiu]

LGTM

[XiaoguangHu01]

LGTM