Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 2 年 前同步成功

通知 2325
Star 20933
Fork 5424
P
Paddle
“0067a2e4ec97563fe907b0079f9e599fe8336c84”上不存在“test/legacy_test/auto_checkpoint_utils.py”
未验证 提交 d30d85da 编写于 作者: C Chen Weihang 提交者： GitHub
浏览文件
操作

[Phi] Replace all prefix PT by PD and fix typo (#40046) 

* replace prefix pt by pd

* replace added kernel

* revert util change

* pd kernel to phi

* resolve conflict

* resolve conflict
上级 7e076e7b
隐藏空白更改
内联 并排
Showing with 350 addition and 349 deletion
paddle/fluid/framework/infershape_utils.h
浏览文件 @ d30d85da
...@@ -29,7 +29,7 @@ namespace framework { ...@@ -29,7 +29,7 @@ namespace framework {
phi::InferMetaContext BuildInferMetaContext(InferShapeContext* ctx, phi::InferMetaContext BuildInferMetaContext(InferShapeContext* ctx,
const std::string& op_type); const std::string& op_type);
#define DELCARE_INFER_SHAPE_FUNCTOR(op_type, functor_name, fn) \ #define DECLARE_INFER_SHAPE_FUNCTOR(op_type, functor_name, fn) \
struct functor_name : public paddle::framework::InferShapeBase { \ struct functor_name : public paddle::framework::InferShapeBase { \
void operator()( \ void operator()( \
paddle::framework::InferShapeContext* ctx) const override { \ paddle::framework::InferShapeContext* ctx) const override { \
......
paddle/fluid/framework/infershape_utils_test.cc
浏览文件 @ d30d85da
...@@ -110,9 +110,9 @@ void InferShapeUtilsTestKernel( ...@@ -110,9 +110,9 @@ void InferShapeUtilsTestKernel(
} // namespace framework } // namespace framework
} // namespace paddle } // namespace paddle
DELCARE_INFER_SHAPE_FUNCTOR(infer_shape_utils_test, DECLARE_INFER_SHAPE_FUNCTOR(infer_shape_utils_test,
InferShapeUtilsTestInferShapeFunctor, InferShapeUtilsTestInferShapeFunctor,
PT_INFER_META(paddle::framework::TestInferMeta)); PD_INFER_META(paddle::framework::TestInferMeta));
REGISTER_OPERATOR(infer_shape_utils_test, REGISTER_OPERATOR(infer_shape_utils_test,
paddle::framework::InferShapeUtilsTestOp, paddle::framework::InferShapeUtilsTestOp,
paddle::framework::InferShapeUtilsTestOpMaker, paddle::framework::InferShapeUtilsTestOpMaker,
......
paddle/fluid/operators/abs_op.cc
浏览文件 @ d30d85da
...@@ -141,8 +141,8 @@ class AbsDoubleGradOp : public framework::OperatorWithKernel { ...@@ -141,8 +141,8 @@ class AbsDoubleGradOp : public framework::OperatorWithKernel {
} // namespace operators } // namespace operators
} // namespace paddle } // namespace paddle
DELCARE_INFER_SHAPE_FUNCTOR(abs, AbsInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(abs, AbsInferShapeFunctor,
PT_INFER_META(phi::UnchangedInferMeta)); PD_INFER_META(phi::UnchangedInferMeta));
namespace ops = paddle::operators; namespace ops = paddle::operators;
......
paddle/fluid/operators/addmm_op.cc
浏览文件 @ d30d85da
...@@ -147,8 +147,8 @@ class AddMMOpGradMaker : public framework::SingleGradOpMaker<T> { ...@@ -147,8 +147,8 @@ class AddMMOpGradMaker : public framework::SingleGradOpMaker<T> {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(addmm, AddmmInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(addmm, AddmmInferShapeFunctor,
PT_INFER_META(phi::AddmmInferMeta)); PD_INFER_META(phi::AddmmInferMeta));
REGISTER_OPERATOR(addmm, ops::AddMMOp, ops::AddMMOpMaker, REGISTER_OPERATOR(addmm, ops::AddMMOp, ops::AddMMOpMaker,
ops::AddMMOpGradMaker<paddle::framework::OpDesc>, ops::AddMMOpGradMaker<paddle::framework::OpDesc>,
ops::AddMMOpGradMaker<paddle::imperative::OpBase>, ops::AddMMOpGradMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/atan2_op.cc
浏览文件 @ d30d85da
...@@ -105,8 +105,8 @@ class Atan2OpVarTypeInference : public framework::VarTypeInference { ...@@ -105,8 +105,8 @@ class Atan2OpVarTypeInference : public framework::VarTypeInference {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(atan2, Atan2InferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(atan2, Atan2InferShapeFunctor,
PT_INFER_META(phi::Atan2InferMeta)); PD_INFER_META(phi::Atan2InferMeta));
REGISTER_OPERATOR(atan2, ops::Atan2Op, ops::Atan2OpMaker, REGISTER_OPERATOR(atan2, ops::Atan2Op, ops::Atan2OpMaker,
ops::Atan2GradMaker<paddle::framework::OpDesc>, ops::Atan2GradMaker<paddle::framework::OpDesc>,
ops::Atan2GradMaker<paddle::imperative::OpBase>, ops::Atan2GradMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/bce_loss_op.cc
浏览文件 @ d30d85da
...@@ -138,8 +138,8 @@ DECLARE_INPLACE_OP_INFERER(BCELossGradInplaceInferer, ...@@ -138,8 +138,8 @@ DECLARE_INPLACE_OP_INFERER(BCELossGradInplaceInferer,
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(bce_loss, BCELossInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(bce_loss, BCELossInferShapeFunctor,
PT_INFER_META(phi::BCELossInferMeta)); PD_INFER_META(phi::BCELossInferMeta));
REGISTER_OPERATOR(bce_loss, ops::BCELossOp, ops::BCELossOpMaker, REGISTER_OPERATOR(bce_loss, ops::BCELossOp, ops::BCELossOpMaker,
ops::BCELossGradOpMaker<paddle::framework::OpDesc>, ops::BCELossGradOpMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/bilinear_tensor_product_op.cc
浏览文件 @ d30d85da
...@@ -90,12 +90,12 @@ class BilinearTensorProductGradOpMaker ...@@ -90,12 +90,12 @@ class BilinearTensorProductGradOpMaker
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(bilinear_tensor_product, DECLARE_INFER_SHAPE_FUNCTOR(bilinear_tensor_product,
BilinearTensorProductInferShapeFunctor, BilinearTensorProductInferShapeFunctor,
PT_INFER_META(phi::BilinearTensorProductInferMeta)); PD_INFER_META(phi::BilinearTensorProductInferMeta));
DELCARE_INFER_SHAPE_FUNCTOR( DECLARE_INFER_SHAPE_FUNCTOR(
bilinear_tensor_product_grad, BilinearTensorProductGradInferShapeFunctor, bilinear_tensor_product_grad, BilinearTensorProductGradInferShapeFunctor,
PT_INFER_META(phi::BilinearTensorProductGradInferMeta)); PD_INFER_META(phi::BilinearTensorProductGradInferMeta));
REGISTER_OPERATOR( REGISTER_OPERATOR(
bilinear_tensor_product, ops::BilinearTensorProductOp, bilinear_tensor_product, ops::BilinearTensorProductOp,
......
paddle/fluid/operators/broadcast_tensors_op.cc
浏览文件 @ d30d85da
...@@ -167,9 +167,9 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(BroadcastTensorsGradNoNeedBufVarsInferer, ...@@ -167,9 +167,9 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(BroadcastTensorsGradNoNeedBufVarsInferer,
namespace ops = paddle::operators; namespace ops = paddle::operators;
namespace plat = paddle::platform; namespace plat = paddle::platform;
DELCARE_INFER_SHAPE_FUNCTOR(broadcast_tensors, DECLARE_INFER_SHAPE_FUNCTOR(broadcast_tensors,
BroadcastTensorsInferShapeFunctor, BroadcastTensorsInferShapeFunctor,
PT_INFER_META(phi::BroadcastTensorsInferMeta)); PD_INFER_META(phi::BroadcastTensorsInferMeta));
REGISTER_OPERATOR(broadcast_tensors, ops::BroadcastTensorsOp, REGISTER_OPERATOR(broadcast_tensors, ops::BroadcastTensorsOp,
ops::BroadcastTensorsOpMaker, ops::BroadcastTensorsOpMaker,
......
paddle/fluid/operators/cholesky_op.cc
浏览文件 @ d30d85da
...@@ -90,8 +90,8 @@ class CholeskyGradOpMaker : public framework::SingleGradOpMaker<T> { ...@@ -90,8 +90,8 @@ class CholeskyGradOpMaker : public framework::SingleGradOpMaker<T> {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(cholesky, CholeskyInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(cholesky, CholeskyInferShapeFunctor,
PT_INFER_META(phi::CholeskyInferMeta)); PD_INFER_META(phi::CholeskyInferMeta));
REGISTER_OPERATOR(cholesky, ops::CholeskyOp, ops::CholeskyOpMaker, REGISTER_OPERATOR(cholesky, ops::CholeskyOp, ops::CholeskyOpMaker,
ops::CholeskyGradOpMaker<paddle::framework::OpDesc>, ops::CholeskyGradOpMaker<paddle::framework::OpDesc>,
ops::CholeskyGradOpMaker<paddle::imperative::OpBase>, ops::CholeskyGradOpMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/concat_op.cc
浏览文件 @ d30d85da
...@@ -205,8 +205,8 @@ class ConcatDoubleGradOpMaker : public framework::SingleGradOpMaker<T> { ...@@ -205,8 +205,8 @@ class ConcatDoubleGradOpMaker : public framework::SingleGradOpMaker<T> {
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(concat, ConcatInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(concat, ConcatInferShapeFunctor,
PT_INFER_META(phi::ConcatInferMeta)); PD_INFER_META(phi::ConcatInferMeta));
REGISTER_OPERATOR(concat, ops::ConcatOp, ops::ConcatOpMaker, REGISTER_OPERATOR(concat, ops::ConcatOp, ops::ConcatOpMaker,
ops::ConcatGradOpMaker<paddle::framework::OpDesc>, ops::ConcatGradOpMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/conj_op.cc
浏览文件 @ d30d85da
...@@ -66,8 +66,8 @@ class ConjGradMaker : public framework::SingleGradOpMaker<T> { ...@@ -66,8 +66,8 @@ class ConjGradMaker : public framework::SingleGradOpMaker<T> {
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(conj, ConjInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(conj, ConjInferShapeFunctor,
PT_INFER_META(phi::UnchangedInferMeta)); PD_INFER_META(phi::UnchangedInferMeta));
REGISTER_OPERATOR(conj, ops::ConjOp, ops::ConjOpMaker, REGISTER_OPERATOR(conj, ops::ConjOp, ops::ConjOpMaker,
ops::ConjGradMaker<paddle::framework::OpDesc>, ops::ConjGradMaker<paddle::framework::OpDesc>,
ops::ConjGradMaker<paddle::imperative::OpBase>, ops::ConjGradMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/controlflow/compare_all_op.cc
浏览文件 @ d30d85da
...@@ -58,8 +58,8 @@ class CompareReduceOp : public framework::OperatorWithKernel { ...@@ -58,8 +58,8 @@ class CompareReduceOp : public framework::OperatorWithKernel {
}; \ }; \
char _##op_type##Comment::type[]{#op_type}; \ char _##op_type##Comment::type[]{#op_type}; \
char _##op_type##Comment::equation[]{_equation}; \ char _##op_type##Comment::equation[]{_equation}; \
DELCARE_INFER_SHAPE_FUNCTOR(op_type, op_type##_InferShapeFunctor, \ DECLARE_INFER_SHAPE_FUNCTOR(op_type, op_type##_InferShapeFunctor, \
PT_INFER_META(phi::CompareAllInferMeta)); \ PD_INFER_META(phi::CompareAllInferMeta)); \
REGISTER_OPERATOR( \ REGISTER_OPERATOR( \
op_type, ::paddle::operators::CompareReduceOp<_##op_type##Comment>, \ op_type, ::paddle::operators::CompareReduceOp<_##op_type##Comment>, \
::paddle::operators::CompareReduceOpProtoMaker<_##op_type##Comment>, \ ::paddle::operators::CompareReduceOpProtoMaker<_##op_type##Comment>, \
......
paddle/fluid/operators/controlflow/compare_op.cc
浏览文件 @ d30d85da
...@@ -96,8 +96,8 @@ class CompareOp : public framework::OperatorWithKernel { ...@@ -96,8 +96,8 @@ class CompareOp : public framework::OperatorWithKernel {
}; \ }; \
char _##op_type##Comment::type[]{#op_type}; \ char _##op_type##Comment::type[]{#op_type}; \
char _##op_type##Comment::equation[]{_equation}; \ char _##op_type##Comment::equation[]{_equation}; \
DELCARE_INFER_SHAPE_FUNCTOR(op_type, op_type##_InferShapeFunctor, \ DECLARE_INFER_SHAPE_FUNCTOR(op_type, op_type##_InferShapeFunctor, \
PT_INFER_META(phi::CompareInferMeta)); \ PD_INFER_META(phi::CompareInferMeta)); \
REGISTER_OPERATOR( \ REGISTER_OPERATOR( \
op_type, ::paddle::operators::CompareOp<_##op_type##Comment>, \ op_type, ::paddle::operators::CompareOp<_##op_type##Comment>, \
::paddle::operators::CompareOpProtoMaker<_##op_type##Comment>, \ ::paddle::operators::CompareOpProtoMaker<_##op_type##Comment>, \
......
paddle/fluid/operators/cross_op.cc
浏览文件 @ d30d85da
...@@ -109,8 +109,8 @@ class CrossGradMaker : public framework::SingleGradOpMaker<T> { ...@@ -109,8 +109,8 @@ class CrossGradMaker : public framework::SingleGradOpMaker<T> {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(cross, CrossInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(cross, CrossInferShapeFunctor,
PT_INFER_META(phi::CrossInferMeta)); PD_INFER_META(phi::CrossInferMeta));
REGISTER_OPERATOR(cross, ops::CrossOp, ops::CrossOpMaker, REGISTER_OPERATOR(cross, ops::CrossOp, ops::CrossOpMaker,
ops::CrossGradMaker<paddle::framework::OpDesc>, ops::CrossGradMaker<paddle::framework::OpDesc>,
ops::CrossGradMaker<paddle::imperative::OpBase>, ops::CrossGradMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/diag_v2_op.cc
浏览文件 @ d30d85da
...@@ -62,8 +62,8 @@ class DiagV2OpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -62,8 +62,8 @@ class DiagV2OpMaker : public framework::OpProtoAndCheckerMaker {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(diag_v2, DiagInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(diag_v2, DiagInferShapeFunctor,
PT_INFER_META(phi::DiagInferMeta)); PD_INFER_META(phi::DiagInferMeta));
REGISTER_OPERATOR( REGISTER_OPERATOR(
diag_v2, ops::DiagV2Op, ops::DiagV2OpMaker, diag_v2, ops::DiagV2Op, ops::DiagV2OpMaker,
......
paddle/fluid/operators/diagonal_op.cc
浏览文件 @ d30d85da
...@@ -105,8 +105,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(DiagonalGradNoNeedBufferVarsInferer, ...@@ -105,8 +105,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(DiagonalGradNoNeedBufferVarsInferer,
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(diagonal, DiagonalInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(diagonal, DiagonalInferShapeFunctor,
PT_INFER_META(phi::DiagonalInferMeta)); PD_INFER_META(phi::DiagonalInferMeta));
REGISTER_OPERATOR(diagonal, ops::DiagonalOp, ops::DiagonalOpMaker, REGISTER_OPERATOR(diagonal, ops::DiagonalOp, ops::DiagonalOpMaker,
ops::DiagonalGradOpMaker<paddle::framework::OpDesc>, ops::DiagonalGradOpMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/dist_op.cc
浏览文件 @ d30d85da
...@@ -124,8 +124,8 @@ class DistGradOpMaker : public framework::SingleGradOpMaker<T> { ...@@ -124,8 +124,8 @@ class DistGradOpMaker : public framework::SingleGradOpMaker<T> {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(dist, DistInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(dist, DistInferShapeFunctor,
PT_INFER_META(phi::DistInferMeta)); PD_INFER_META(phi::DistInferMeta));
REGISTER_OPERATOR(dist, ops::DistOp, ops::DistOpMaker, REGISTER_OPERATOR(dist, ops::DistOp, ops::DistOpMaker,
ops::DistGradOpMaker<paddle::framework::OpDesc>, ops::DistGradOpMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/dot_op.cc
浏览文件 @ d30d85da
...@@ -101,8 +101,8 @@ class DotOpGradMaker : public framework::SingleGradOpMaker<T> { ...@@ -101,8 +101,8 @@ class DotOpGradMaker : public framework::SingleGradOpMaker<T> {
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(dot, DotInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(dot, DotInferShapeFunctor,
PT_INFER_META(phi::DotInferMeta)); PD_INFER_META(phi::DotInferMeta));
REGISTER_OPERATOR(dot, ops::DotOp, ops::DotOpMaker, REGISTER_OPERATOR(dot, ops::DotOp, ops::DotOpMaker,
ops::DotOpGradMaker<paddle::framework::OpDesc>, ops::DotOpGradMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/empty_op.cc
浏览文件 @ d30d85da
...@@ -88,8 +88,8 @@ class EmptyOpVarTypeInference : public framework::VarTypeInference { ...@@ -88,8 +88,8 @@ class EmptyOpVarTypeInference : public framework::VarTypeInference {
namespace ops = paddle::operators; namespace ops = paddle::operators;
namespace plat = paddle::platform; namespace plat = paddle::platform;
DELCARE_INFER_SHAPE_FUNCTOR(empty, EmptyInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(empty, EmptyInferShapeFunctor,
PT_INFER_META(phi::CreateInferMeta)); PD_INFER_META(phi::CreateInferMeta));
REGISTER_OP_WITHOUT_GRADIENT(empty, ops::EmptyOp, ops::EmptyOpMaker, REGISTER_OP_WITHOUT_GRADIENT(empty, ops::EmptyOp, ops::EmptyOpMaker,
ops::EmptyOpVarTypeInference, ops::EmptyOpVarTypeInference,
EmptyInferShapeFunctor); EmptyInferShapeFunctor);
paddle/fluid/operators/erfinv_op.cc
浏览文件 @ d30d85da
...@@ -73,8 +73,8 @@ DECLARE_INPLACE_OP_INFERER(ErfinvInplaceInferer, {"X", "Out"}); ...@@ -73,8 +73,8 @@ DECLARE_INPLACE_OP_INFERER(ErfinvInplaceInferer, {"X", "Out"});
} // namespace operators } // namespace operators
} // namespace paddle } // namespace paddle
DELCARE_INFER_SHAPE_FUNCTOR(erfinv, ErfinvInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(erfinv, ErfinvInferShapeFunctor,
PT_INFER_META(phi::UnchangedInferMeta)); PD_INFER_META(phi::UnchangedInferMeta));
REGISTER_OPERATOR( REGISTER_OPERATOR(
erfinv, paddle::operators::ErfinvOp, paddle::operators::ErfinvOpMaker, erfinv, paddle::operators::ErfinvOp, paddle::operators::ErfinvOpMaker,
......
paddle/fluid/operators/eye_op.cc
浏览文件 @ d30d85da
...@@ -67,8 +67,8 @@ Return an identity tensor whose shape is [num_rows, num_columns]. ...@@ -67,8 +67,8 @@ Return an identity tensor whose shape is [num_rows, num_columns].
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(eye, EyeInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(eye, EyeInferShapeFunctor,
PT_INFER_META(phi::EyeInferMeta)); PD_INFER_META(phi::EyeInferMeta));
REGISTER_OPERATOR( REGISTER_OPERATOR(
eye, ops::EyeOp, ops::EyeOpMaker, ops::EyeOpVarTypeInference, eye, ops::EyeOp, ops::EyeOpMaker, ops::EyeOpVarTypeInference,
......
paddle/fluid/operators/gather_nd_op.cc
浏览文件 @ d30d85da
...@@ -130,11 +130,11 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(GatherNdGradNoNeedBufferVarInferer, "X"); ...@@ -130,11 +130,11 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(GatherNdGradNoNeedBufferVarInferer, "X");
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(gather_nd, GatherNdInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(gather_nd, GatherNdInferShapeFunctor,
PT_INFER_META(phi::GatherNdInferMeta)); PD_INFER_META(phi::GatherNdInferMeta));
DELCARE_INFER_SHAPE_FUNCTOR(gather_nd_grad, GatherNdGradInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(gather_nd_grad, GatherNdGradInferShapeFunctor,
PT_INFER_META(phi::GatherNdGradInferMeta)); PD_INFER_META(phi::GatherNdGradInferMeta));
REGISTER_OPERATOR(gather_nd, ops::GatherNdOp, ops::GatherNdOpMaker, REGISTER_OPERATOR(gather_nd, ops::GatherNdOp, ops::GatherNdOpMaker,
ops::GatherNdGradOpMaker<paddle::framework::OpDesc>, ops::GatherNdGradOpMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/gather_tree_op.cc
浏览文件 @ d30d85da
...@@ -61,8 +61,8 @@ selected ids. ...@@ -61,8 +61,8 @@ selected ids.
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(gather_tree, GatherTreeInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(gather_tree, GatherTreeInferShapeFunctor,
PT_INFER_META(phi::GatherTreeMeta)); PD_INFER_META(phi::GatherTreeMeta));
REGISTER_OPERATOR(gather_tree, ops::GatherTreeOp, ops::GatherTreeOpMaker, REGISTER_OPERATOR(gather_tree, ops::GatherTreeOp, ops::GatherTreeOpMaker,
GatherTreeInferShapeFunctor); GatherTreeInferShapeFunctor);
paddle/fluid/operators/gumbel_softmax_op.cc
浏览文件 @ d30d85da
...@@ -90,11 +90,11 @@ class GumbelSoftmaxGradOpMaker : public framework::SingleGradOpMaker<T> { ...@@ -90,11 +90,11 @@ class GumbelSoftmaxGradOpMaker : public framework::SingleGradOpMaker<T> {
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(gumbel_softmax, GumbelSoftmaxInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(gumbel_softmax, GumbelSoftmaxInferShapeFunctor,
PT_INFER_META(phi::GumbelSoftmaxInferMeta)); PD_INFER_META(phi::GumbelSoftmaxInferMeta));
DELCARE_INFER_SHAPE_FUNCTOR(gumbel_softmax_grad, DECLARE_INFER_SHAPE_FUNCTOR(gumbel_softmax_grad,
GumbelSoftmaxGradInferShapeFunctor, GumbelSoftmaxGradInferShapeFunctor,
PT_INFER_META(phi::GumbelSoftmaxGradInferMeta)); PD_INFER_META(phi::GumbelSoftmaxGradInferMeta));
REGISTER_OPERATOR(gumbel_softmax, ops::GumbelSoftmaxOp, REGISTER_OPERATOR(gumbel_softmax, ops::GumbelSoftmaxOp,
ops::GumbelSoftmaxOpMaker, ops::GumbelSoftmaxOpMaker,
......
paddle/fluid/operators/huber_loss_op.cc
浏览文件 @ d30d85da
...@@ -112,8 +112,8 @@ class HuberLossGradOpMaker : public framework::SingleGradOpMaker<T> { ...@@ -112,8 +112,8 @@ class HuberLossGradOpMaker : public framework::SingleGradOpMaker<T> {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(huber_loss, HuberLossInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(huber_loss, HuberLossInferShapeFunctor,
PT_INFER_META(phi::HuberLossInferMeta)); PD_INFER_META(phi::HuberLossInferMeta));
REGISTER_OPERATOR(huber_loss, ops::HuberLossOp, ops::HuberLossOpMaker<float>, REGISTER_OPERATOR(huber_loss, ops::HuberLossOp, ops::HuberLossOpMaker<float>,
ops::HuberLossGradOpMaker<paddle::framework::OpDesc>, ops::HuberLossGradOpMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/imag_op.cc
浏览文件 @ d30d85da
...@@ -82,8 +82,8 @@ DECLARE_INPLACE_OP_INFERER(ImagGradOpInplaceInferer, ...@@ -82,8 +82,8 @@ DECLARE_INPLACE_OP_INFERER(ImagGradOpInplaceInferer,
} // namespace operators } // namespace operators
} // namespace paddle } // namespace paddle
DELCARE_INFER_SHAPE_FUNCTOR(imag, ImagInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(imag, ImagInferShapeFunctor,
PT_INFER_META(phi::RealAndImagInferMeta)); PD_INFER_META(phi::RealAndImagInferMeta));
namespace ops = paddle::operators; namespace ops = paddle::operators;
......
paddle/fluid/operators/increment_op.cc
浏览文件 @ d30d85da
...@@ -87,8 +87,8 @@ class IncrementGradOpMaker : public framework::SingleGradOpMaker<T> { ...@@ -87,8 +87,8 @@ class IncrementGradOpMaker : public framework::SingleGradOpMaker<T> {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(increment, IncrementInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(increment, IncrementInferShapeFunctor,
PT_INFER_META(phi::IncrementInferMeta)); PD_INFER_META(phi::IncrementInferMeta));
REGISTER_OPERATOR(increment, ops::IncrementOp, ops::IncrementOpMaker, REGISTER_OPERATOR(increment, ops::IncrementOp, ops::IncrementOpMaker,
ops::IncrementGradOpMaker<paddle::framework::OpDesc>, ops::IncrementGradOpMaker<paddle::framework::OpDesc>,
ops::IncrementGradOpMaker<paddle::imperative::OpBase>, ops::IncrementGradOpMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/index_sample_op.cc
浏览文件 @ d30d85da
...@@ -100,8 +100,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(IndexSampleGradNoNeedBufferVarInferer, "X"); ...@@ -100,8 +100,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(IndexSampleGradNoNeedBufferVarInferer, "X");
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(index_sample, IndexSampleInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(index_sample, IndexSampleInferShapeFunctor,
PT_INFER_META(phi::IndexSampleInferMeta)); PD_INFER_META(phi::IndexSampleInferMeta));
REGISTER_OPERATOR(index_sample, ops::IndexSampleOp, ops::IndexSampleOpMaker, REGISTER_OPERATOR(index_sample, ops::IndexSampleOp, ops::IndexSampleOpMaker,
ops::IndexSampleGradMaker<paddle::framework::OpDesc>, ops::IndexSampleGradMaker<paddle::framework::OpDesc>,
ops::IndexSampleGradMaker<paddle::imperative::OpBase>, ops::IndexSampleGradMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/lerp_op.cc
浏览文件 @ d30d85da
...@@ -85,8 +85,8 @@ DECLARE_INPLACE_OP_INFERER(LerpInplaceInferer, {"X", "Out"}); ...@@ -85,8 +85,8 @@ DECLARE_INPLACE_OP_INFERER(LerpInplaceInferer, {"X", "Out"});
} // namespace operators } // namespace operators
} // namespace paddle } // namespace paddle
DELCARE_INFER_SHAPE_FUNCTOR(lerp, LerpInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(lerp, LerpInferShapeFunctor,
PT_INFER_META(phi::LerpInferMeta)); PD_INFER_META(phi::LerpInferMeta));
REGISTER_OPERATOR( REGISTER_OPERATOR(
lerp, paddle::operators::LerpOp, paddle::operators::LerpOpMaker, lerp, paddle::operators::LerpOp, paddle::operators::LerpOpMaker,
paddle::operators::LerpOpGradMaker<paddle::framework::OpDesc>, paddle::operators::LerpOpGradMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/matmul_v2_op.cc
浏览文件 @ d30d85da
...@@ -524,8 +524,8 @@ REGISTER_OPERATOR(matmul_v2, ops::MatMulV2Op, ops::MatMulV2OpMaker, ...@@ -524,8 +524,8 @@ REGISTER_OPERATOR(matmul_v2, ops::MatMulV2Op, ops::MatMulV2OpMaker,
ops::MatMulV2GradOpMaker<paddle::framework::OpDesc>, ops::MatMulV2GradOpMaker<paddle::framework::OpDesc>,
ops::MatMulV2GradOpMaker<paddle::imperative::OpBase>); ops::MatMulV2GradOpMaker<paddle::imperative::OpBase>);
DELCARE_INFER_SHAPE_FUNCTOR(matmul_v2_grad, MatMulV2GradInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(matmul_v2_grad, MatMulV2GradInferShapeFunctor,
PT_INFER_META(phi::GeneralBinaryGradInferMeta)); PD_INFER_META(phi::GeneralBinaryGradInferMeta));
REGISTER_OPERATOR(matmul_v2_grad, ops::MatMulV2OpGrad, REGISTER_OPERATOR(matmul_v2_grad, ops::MatMulV2OpGrad,
ops::MatMulV2OpDoubleGradMaker<paddle::framework::OpDesc>, ops::MatMulV2OpDoubleGradMaker<paddle::framework::OpDesc>,
ops::MatMulV2OpDoubleGradMaker<paddle::imperative::OpBase>, ops::MatMulV2OpDoubleGradMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/multinomial_op.cc
浏览文件 @ d30d85da
...@@ -53,8 +53,8 @@ class MultinomialOp : public framework::OperatorWithKernel { ...@@ -53,8 +53,8 @@ class MultinomialOp : public framework::OperatorWithKernel {
namespace ops = paddle::operators; namespace ops = paddle::operators;
namespace plat = paddle::platform; namespace plat = paddle::platform;
DELCARE_INFER_SHAPE_FUNCTOR(multinomial, MultinomialInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(multinomial, MultinomialInferShapeFunctor,
PT_INFER_META(phi::MultinomialInferMeta)); PD_INFER_META(phi::MultinomialInferMeta));
REGISTER_OPERATOR( REGISTER_OPERATOR(
multinomial, ops::MultinomialOp, ops::MultinomialOpMaker, multinomial, ops::MultinomialOp, ops::MultinomialOpMaker,
paddle::framework::EmptyGradOpMaker<paddle::framework::OpDesc>, paddle::framework::EmptyGradOpMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/mv_op.cc
浏览文件 @ d30d85da
...@@ -94,8 +94,8 @@ class MVOpGrad : public framework::OperatorWithKernel { ...@@ -94,8 +94,8 @@ class MVOpGrad : public framework::OperatorWithKernel {
namespace ops = paddle::operators; namespace ops = paddle::operators;
namespace plat = paddle::platform; namespace plat = paddle::platform;
DELCARE_INFER_SHAPE_FUNCTOR(mv, MvInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(mv, MvInferShapeFunctor,
PT_INFER_META(phi::MvInferMeta)); PD_INFER_META(phi::MvInferMeta));
REGISTER_OPERATOR(mv, ops::MVOp, ops::MVOpMaker, REGISTER_OPERATOR(mv, ops::MVOp, ops::MVOpMaker,
ops::MVOpGradMaker<paddle::framework::OpDesc>, ops::MVOpGradMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/pixel_shuffle_op.cc
浏览文件 @ d30d85da
...@@ -124,8 +124,8 @@ class PixelShuffleGradOp : public framework::OperatorWithKernel { ...@@ -124,8 +124,8 @@ class PixelShuffleGradOp : public framework::OperatorWithKernel {
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(pixel_shuffle, PixelShuffleInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(pixel_shuffle, PixelShuffleInferShapeFunctor,
PT_INFER_META(phi::PixelShuffleInferMeta)); PD_INFER_META(phi::PixelShuffleInferMeta));
REGISTER_OPERATOR(pixel_shuffle, ops::PixelShuffleOp, ops::PixelShuffleOpMaker, REGISTER_OPERATOR(pixel_shuffle, ops::PixelShuffleOp, ops::PixelShuffleOpMaker,
ops::PixelShuffleGradMaker<paddle::framework::OpDesc>, ops::PixelShuffleGradMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/poisson_op.cc
浏览文件 @ d30d85da
...@@ -87,8 +87,8 @@ class PoissonGradOpMaker : public framework::SingleGradOpMaker<T> { ...@@ -87,8 +87,8 @@ class PoissonGradOpMaker : public framework::SingleGradOpMaker<T> {
namespace ops = paddle::operators; namespace ops = paddle::operators;
namespace plat = paddle::platform; namespace plat = paddle::platform;
DELCARE_INFER_SHAPE_FUNCTOR(poisson, PoissonInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(poisson, PoissonInferShapeFunctor,
PT_INFER_META(phi::UnchangedInferMeta)); PD_INFER_META(phi::UnchangedInferMeta));
REGISTER_OPERATOR(poisson, ops::PoissonOp, ops::PoissonOpMaker, REGISTER_OPERATOR(poisson, ops::PoissonOp, ops::PoissonOpMaker,
ops::PoissonOpInferVarType, ops::PoissonOpInferVarType,
......
paddle/fluid/operators/real_op.cc
浏览文件 @ d30d85da
...@@ -82,8 +82,8 @@ DECLARE_INPLACE_OP_INFERER(RealGradOpInplaceInferer, ...@@ -82,8 +82,8 @@ DECLARE_INPLACE_OP_INFERER(RealGradOpInplaceInferer,
} // namespace operators } // namespace operators
} // namespace paddle } // namespace paddle
DELCARE_INFER_SHAPE_FUNCTOR(real, RealInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(real, RealInferShapeFunctor,
PT_INFER_META(phi::RealAndImagInferMeta)); PD_INFER_META(phi::RealAndImagInferMeta));
namespace ops = paddle::operators; namespace ops = paddle::operators;
......
paddle/fluid/operators/reduce_ops/reduce_mean_op.cc
浏览文件 @ d30d85da
...@@ -96,8 +96,8 @@ class __reduce_meanMaker__ : public ops::ReduceOpMaker { ...@@ -96,8 +96,8 @@ class __reduce_meanMaker__ : public ops::ReduceOpMaker {
virtual std::string GetOpType() const { return "Reduce reduce_mean"; } virtual std::string GetOpType() const { return "Reduce reduce_mean"; }
}; };
DELCARE_INFER_SHAPE_FUNCTOR(reduce_mean, ReduceMeanInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(reduce_mean, ReduceMeanInferShapeFunctor,
PT_INFER_META(phi::MeanRawInferMeta)); PD_INFER_META(phi::MeanRawInferMeta));
REGISTER_OPERATOR(reduce_mean, ops::ReduceOp, __reduce_meanMaker__, REGISTER_OPERATOR(reduce_mean, ops::ReduceOp, __reduce_meanMaker__,
ops::ReduceMeanOpGradMaker<paddle::framework::OpDesc>, ops::ReduceMeanOpGradMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/reduce_ops/reduce_sum_op.cc
浏览文件 @ d30d85da
...@@ -102,8 +102,8 @@ class ReduceSumOpMaker : public ops::ReduceOpMaker { ...@@ -102,8 +102,8 @@ class ReduceSumOpMaker : public ops::ReduceOpMaker {
virtual std::string GetOpType() const { return "Reduce reduce_sum"; } virtual std::string GetOpType() const { return "Reduce reduce_sum"; }
}; };
DELCARE_INFER_SHAPE_FUNCTOR(reduce_sum, ReduceSumInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(reduce_sum, ReduceSumInferShapeFunctor,
PT_INFER_META(phi::ReduceInferMetaBase)); PD_INFER_META(phi::ReduceInferMetaBase));
REGISTER_OPERATOR(reduce_sum, ops::ReduceOp, ReduceSumOpMaker, REGISTER_OPERATOR(reduce_sum, ops::ReduceOp, ReduceSumOpMaker,
ops::ReduceSumVarTypeInference, ops::ReduceSumVarTypeInference,
......
paddle/fluid/operators/scale_op.cc
浏览文件 @ d30d85da
...@@ -121,8 +121,8 @@ DECLARE_INPLACE_OP_INFERER(ScaleOpInplaceInferer, {"X", "Out"}); ...@@ -121,8 +121,8 @@ DECLARE_INPLACE_OP_INFERER(ScaleOpInplaceInferer, {"X", "Out"});
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(scale, ScaleInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(scale, ScaleInferShapeFunctor,
PT_INFER_META(phi::UnchangedInferMeta)); PD_INFER_META(phi::UnchangedInferMeta));
REGISTER_OPERATOR(scale, ops::ScaleOp, ops::ScaleOpMaker, REGISTER_OPERATOR(scale, ops::ScaleOp, ops::ScaleOpMaker,
ops::ScaleGradMaker<paddle::framework::OpDesc>, ops::ScaleGradMaker<paddle::framework::OpDesc>,
ops::ScaleGradMaker<paddle::imperative::OpBase>, ops::ScaleGradMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/scatter_nd_add_op.cc
浏览文件 @ d30d85da
...@@ -119,12 +119,12 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(ScatterNdAddGradNoNeedBufferVarsInferer, ...@@ -119,12 +119,12 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(ScatterNdAddGradNoNeedBufferVarsInferer,
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(scatter_nd_add, ScatterNdAddInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(scatter_nd_add, ScatterNdAddInferShapeFunctor,
PT_INFER_META(phi::ScatterNdAddInferMeta)); PD_INFER_META(phi::ScatterNdAddInferMeta));
DELCARE_INFER_SHAPE_FUNCTOR(scatter_nd_add_grad, DECLARE_INFER_SHAPE_FUNCTOR(scatter_nd_add_grad,
ScatterNdAddGradInferShapeFunctor, ScatterNdAddGradInferShapeFunctor,
PT_INFER_META(phi::ScatterNdAddGradInferMeta)); PD_INFER_META(phi::ScatterNdAddGradInferMeta));
REGISTER_OPERATOR(scatter_nd_add, ops::ScatterNdAddOp, ops::ScatterNdAddOpMaker, REGISTER_OPERATOR(scatter_nd_add, ops::ScatterNdAddOp, ops::ScatterNdAddOpMaker,
ops::ScatterNdAddGradMaker<paddle::framework::OpDesc>, ops::ScatterNdAddGradMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/scatter_op.cc
浏览文件 @ d30d85da
...@@ -103,11 +103,11 @@ DECLARE_INPLACE_OP_INFERER(ScatterInplaceInferer, {"X", "Out"}); ...@@ -103,11 +103,11 @@ DECLARE_INPLACE_OP_INFERER(ScatterInplaceInferer, {"X", "Out"});
} // namespace operators } // namespace operators
} // namespace paddle } // namespace paddle
DELCARE_INFER_SHAPE_FUNCTOR(scatter, ScatterInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(scatter, ScatterInferShapeFunctor,
PT_INFER_META(phi::ScatterInferMeta)); PD_INFER_META(phi::ScatterInferMeta));
DELCARE_INFER_SHAPE_FUNCTOR(scatter_grad, ScatterGradInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(scatter_grad, ScatterGradInferShapeFunctor,
PT_INFER_META(phi::ScatterGradInferMeta)); PD_INFER_META(phi::ScatterGradInferMeta));
namespace ops = paddle::operators; namespace ops = paddle::operators;
REGISTER_OPERATOR(scatter, ops::ScatterOp, ops::ScatterOpMaker, REGISTER_OPERATOR(scatter, ops::ScatterOp, ops::ScatterOpMaker,
......
paddle/fluid/operators/selu_op.cc
浏览文件 @ d30d85da
...@@ -120,8 +120,8 @@ class SeluGradOp : public framework::OperatorWithKernel { ...@@ -120,8 +120,8 @@ class SeluGradOp : public framework::OperatorWithKernel {
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(selu, SeluInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(selu, SeluInferShapeFunctor,
PT_INFER_META(phi::UnchangedInferMeta)); PD_INFER_META(phi::UnchangedInferMeta));
REGISTER_OPERATOR(selu, ops::SeluOp, ops::SeluOpMaker, ops::SeluOpInferVarType, REGISTER_OPERATOR(selu, ops::SeluOp, ops::SeluOpMaker, ops::SeluOpInferVarType,
ops::SeluGradMaker<paddle::framework::OpDesc>, ops::SeluGradMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/sign_op.cc
浏览文件 @ d30d85da
...@@ -60,8 +60,8 @@ class SignGradMaker : public framework::SingleGradOpMaker<T> { ...@@ -60,8 +60,8 @@ class SignGradMaker : public framework::SingleGradOpMaker<T> {
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(sign, SignInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(sign, SignInferShapeFunctor,
PT_INFER_META(phi::UnchangedInferMeta)); PD_INFER_META(phi::UnchangedInferMeta));
REGISTER_OPERATOR(sign, ops::SignOp, ops::SignOpMaker<float>, REGISTER_OPERATOR(sign, ops::SignOp, ops::SignOpMaker<float>,
ops::SignGradMaker<paddle::framework::OpDesc>, ops::SignGradMaker<paddle::framework::OpDesc>,
ops::SignGradMaker<paddle::imperative::OpBase>, ops::SignGradMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/size_op.cc
浏览文件 @ d30d85da
...@@ -44,8 +44,8 @@ Return the number of elements in the input. ...@@ -44,8 +44,8 @@ Return the number of elements in the input.
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(size, SizeInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(size, SizeInferShapeFunctor,
PT_INFER_META(phi::SizeInferMeta)); PD_INFER_META(phi::SizeInferMeta));
REGISTER_OPERATOR( REGISTER_OPERATOR(
size, ops::SizeOp, ops::SizeOpMaker, size, ops::SizeOp, ops::SizeOpMaker,
paddle::framework::EmptyGradOpMaker<paddle::framework::OpDesc>, paddle::framework::EmptyGradOpMaker<paddle::framework::OpDesc>,
......
paddle/fluid/operators/trace_op.cc
浏览文件 @ d30d85da
...@@ -107,8 +107,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(TraceGradNoNeedBufferVarsInferer, "Input"); ...@@ -107,8 +107,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(TraceGradNoNeedBufferVarsInferer, "Input");
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(trace, TraceInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(trace, TraceInferShapeFunctor,
PT_INFER_META(phi::TraceInferMeta)); PD_INFER_META(phi::TraceInferMeta));
REGISTER_OPERATOR(trace, ops::TraceOp, ops::TraceOpMaker, REGISTER_OPERATOR(trace, ops::TraceOp, ops::TraceOpMaker,
ops::TraceGradOpMaker<paddle::framework::OpDesc>, ops::TraceGradOpMaker<paddle::framework::OpDesc>,
ops::TraceGradOpMaker<paddle::imperative::OpBase>, ops::TraceGradOpMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/triangular_solve_op.cc
浏览文件 @ d30d85da
...@@ -120,8 +120,8 @@ class TriangularSolveOpGradMaker : public framework::SingleGradOpMaker<T> { ...@@ -120,8 +120,8 @@ class TriangularSolveOpGradMaker : public framework::SingleGradOpMaker<T> {
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(triangular_solve, TriangularSolveInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(triangular_solve, TriangularSolveInferShapeFunctor,
PT_INFER_META(phi::TriangularSolveInferMeta)); PD_INFER_META(phi::TriangularSolveInferMeta));
REGISTER_OPERATOR(triangular_solve, ops::TriangularSolveOp, REGISTER_OPERATOR(triangular_solve, ops::TriangularSolveOp,
ops::TriangularSolveOpMaker, ops::TriangularSolveOpMaker,
......
paddle/fluid/operators/trunc_op.cc
浏览文件 @ d30d85da
...@@ -69,8 +69,8 @@ class TruncGradOpMaker : public framework::SingleGradOpMaker<T> { ...@@ -69,8 +69,8 @@ class TruncGradOpMaker : public framework::SingleGradOpMaker<T> {
} // namespace operators } // namespace operators
} // namespace paddle } // namespace paddle
DELCARE_INFER_SHAPE_FUNCTOR(trunc, TruncInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(trunc, TruncInferShapeFunctor,
PT_INFER_META(phi::UnchangedInferMeta)); PD_INFER_META(phi::UnchangedInferMeta));
namespace ops = paddle::operators; namespace ops = paddle::operators;
REGISTER_OPERATOR(trunc, ops::TruncOp, ops::TruncOpMaker, REGISTER_OPERATOR(trunc, ops::TruncOp, ops::TruncOpMaker,
......
paddle/fluid/operators/unfold_op.cc
浏览文件 @ d30d85da
...@@ -119,8 +119,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(UnfoldGradOpNoNeedBufferVarsInferer, "X"); ...@@ -119,8 +119,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(UnfoldGradOpNoNeedBufferVarsInferer, "X");
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(unfold, UnfoldInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(unfold, UnfoldInferShapeFunctor,
PT_INFER_META(phi::UnfoldInferMeta)); PD_INFER_META(phi::UnfoldInferMeta));
REGISTER_OPERATOR(unfold, ops::UnfoldOp, ops::UnfoldOpMaker, REGISTER_OPERATOR(unfold, ops::UnfoldOp, ops::UnfoldOpMaker,
ops::UnfoldGradMaker<paddle::framework::OpDesc>, ops::UnfoldGradMaker<paddle::framework::OpDesc>,
ops::UnfoldGradMaker<paddle::imperative::OpBase>, ops::UnfoldGradMaker<paddle::imperative::OpBase>,
......
paddle/fluid/operators/where_op.cc
浏览文件 @ d30d85da
...@@ -117,8 +117,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(WhereGradNoNeedBufferVarsInferer, "X", "Y"); ...@@ -117,8 +117,8 @@ DECLARE_NO_NEED_BUFFER_VARS_INFERER(WhereGradNoNeedBufferVarsInferer, "X", "Y");
} // namespace paddle } // namespace paddle
namespace ops = paddle::operators; namespace ops = paddle::operators;
DELCARE_INFER_SHAPE_FUNCTOR(where, WhereInferShapeFunctor, DECLARE_INFER_SHAPE_FUNCTOR(where, WhereInferShapeFunctor,
PT_INFER_META(phi::WhereInferMeta)); PD_INFER_META(phi::WhereInferMeta));
REGISTER_OPERATOR(where, ops::WhereOp, ops::WhereOpMaker, REGISTER_OPERATOR(where, ops::WhereOp, ops::WhereOpMaker,
ops::WhereOpGradMaker<paddle::framework::OpDesc>, ops::WhereOpGradMaker<paddle::framework::OpDesc>,
ops::WhereOpGradMaker<paddle::imperative::OpBase>, ops::WhereOpGradMaker<paddle::imperative::OpBase>,
......
paddle/phi/common/data_type.h
浏览文件 @ d30d85da
...@@ -82,7 +82,7 @@ inline size_t SizeOf(DataType data_type) { ...@@ -82,7 +82,7 @@ inline size_t SizeOf(DataType data_type) {
return 0; return 0;
} }
#define PT_FOR_EACH_DATA_TYPE(_) \ #define PD_FOR_EACH_DATA_TYPE(_) \
_(bool, DataType::BOOL) \ _(bool, DataType::BOOL) \
_(int8_t, DataType::INT8) \ _(int8_t, DataType::INT8) \
_(uint8_t, DataType::UINT8) \ _(uint8_t, DataType::UINT8) \
...@@ -105,25 +105,25 @@ struct DataTypeToCppType; ...@@ -105,25 +105,25 @@ struct DataTypeToCppType;
template <typename T> template <typename T>
struct CppTypeToDataType; struct CppTypeToDataType;
#define PT_SPECIALIZE_DataTypeToCppType(cpp_type, data_type) \ #define PD_SPECIALIZE_DataTypeToCppType(cpp_type, data_type) \
template <> \ template <> \
struct DataTypeToCppType<data_type> { \ struct DataTypeToCppType<data_type> { \
using type = cpp_type; \ using type = cpp_type; \
}; };
PT_FOR_EACH_DATA_TYPE(PT_SPECIALIZE_DataTypeToCppType) PD_FOR_EACH_DATA_TYPE(PD_SPECIALIZE_DataTypeToCppType)
#undef PT_SPECIALIZE_DataTypeToCppType #undef PD_SPECIALIZE_DataTypeToCppType
#define PT_SPECIALIZE_CppTypeToDataType(cpp_type, data_type) \ #define PD_SPECIALIZE_CppTypeToDataType(cpp_type, data_type) \
template <> \ template <> \
struct CppTypeToDataType<cpp_type> { \ struct CppTypeToDataType<cpp_type> { \
constexpr static DataType Type() { return data_type; } \ constexpr static DataType Type() { return data_type; } \
}; };
PT_FOR_EACH_DATA_TYPE(PT_SPECIALIZE_CppTypeToDataType) PD_FOR_EACH_DATA_TYPE(PD_SPECIALIZE_CppTypeToDataType)
#undef PT_SPECIALIZE_CppTypeToDataType #undef PD_SPECIALIZE_CppTypeToDataType
inline std::ostream& operator<<(std::ostream& os, DataType dtype) { inline std::ostream& operator<<(std::ostream& os, DataType dtype) {
switch (dtype) { switch (dtype) {
......
paddle/phi/core/compat/op_utils.h
浏览文件 @ d30d85da
...@@ -166,7 +166,7 @@ struct ArgumentMappingFnRegistrar { ...@@ -166,7 +166,7 @@ struct ArgumentMappingFnRegistrar {
}; };
#define PD_REGISTER_BASE_KERNEL_NAME(op_type, base_kernel_name) \ #define PD_REGISTER_BASE_KERNEL_NAME(op_type, base_kernel_name) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PD_STATIC_ASSERT_GLOBAL_NAMESPACE( \
PD_REGISTER_base_kernel_name_ns_check_##op_type, \ PD_REGISTER_base_kernel_name_ns_check_##op_type, \
"PD_REGISTER_BASE_KERNEL_NAME must be called in global namespace."); \ "PD_REGISTER_BASE_KERNEL_NAME must be called in global namespace."); \
static const ::phi::BaseKernelNameRegistrar \ static const ::phi::BaseKernelNameRegistrar \
...@@ -174,7 +174,7 @@ struct ArgumentMappingFnRegistrar { ...@@ -174,7 +174,7 @@ struct ArgumentMappingFnRegistrar {
int TouchBaseKernelNameSymbol_##op_type() { return 0; } int TouchBaseKernelNameSymbol_##op_type() { return 0; }
#define PD_DECLARE_BASE_KERNEL_NAME(op_type) \ #define PD_DECLARE_BASE_KERNEL_NAME(op_type) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PD_STATIC_ASSERT_GLOBAL_NAMESPACE( \
PD_DECLARE_ai_name_ns_check_##op_type, \ PD_DECLARE_ai_name_ns_check_##op_type, \
"PD_DECLARE_BASE_KERNEL_NAME must be called in global namespace."); \ "PD_DECLARE_BASE_KERNEL_NAME must be called in global namespace."); \
extern int TouchBaseKernelNameSymbol_##op_type(); \ extern int TouchBaseKernelNameSymbol_##op_type(); \
...@@ -182,7 +182,7 @@ struct ArgumentMappingFnRegistrar { ...@@ -182,7 +182,7 @@ struct ArgumentMappingFnRegistrar {
TouchBaseKernelNameSymbol_##op_type() TouchBaseKernelNameSymbol_##op_type()
#define PD_REGISTER_ARG_MAPPING_FN(op_type, arg_mapping_fn) \ #define PD_REGISTER_ARG_MAPPING_FN(op_type, arg_mapping_fn) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PD_STATIC_ASSERT_GLOBAL_NAMESPACE( \
PD_REGISTER_arg_map_fn_ns_check_##op_type, \ PD_REGISTER_arg_map_fn_ns_check_##op_type, \
"PD_REGISTER_ARG_MAPPING_FN must be called in global namespace."); \ "PD_REGISTER_ARG_MAPPING_FN must be called in global namespace."); \
static const ::phi::ArgumentMappingFnRegistrar \ static const ::phi::ArgumentMappingFnRegistrar \
...@@ -190,7 +190,7 @@ struct ArgumentMappingFnRegistrar { ...@@ -190,7 +190,7 @@ struct ArgumentMappingFnRegistrar {
int TouchArgumentMappingFnSymbol_##op_type() { return 0; } int TouchArgumentMappingFnSymbol_##op_type() { return 0; }
#define PD_DECLARE_ARG_MAPPING_FN(op_type) \ #define PD_DECLARE_ARG_MAPPING_FN(op_type) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PD_STATIC_ASSERT_GLOBAL_NAMESPACE( \
PD_DECLARE_arg_map_fn_ns_check_##op_type, \ PD_DECLARE_arg_map_fn_ns_check_##op_type, \
"PD_DECLARE_ARG_MAPPING_FN must be called in global namespace."); \ "PD_DECLARE_ARG_MAPPING_FN must be called in global namespace."); \
extern int TouchArgumentMappingFnSymbol_##op_type(); \ extern int TouchArgumentMappingFnSymbol_##op_type(); \
......
paddle/phi/core/infermeta_utils.h
浏览文件 @ d30d85da
...@@ -86,10 +86,10 @@ class InferMetaContext { ...@@ -86,10 +86,10 @@ class InferMetaContext {
paddle::SmallVector<std::pair<int, int>> output_range_; paddle::SmallVector<std::pair<int, int>> output_range_;
}; };
#define PT_INFER_META(...) \ #define PD_INFER_META(...) \
::phi::InferMetaFnImpl<decltype(&__VA_ARGS__), &__VA_ARGS__>::Call ::phi::InferMetaFnImpl<decltype(&__VA_ARGS__), &__VA_ARGS__>::Call
#define PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(attr_type) \ #define PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(attr_type) \
template <typename... Tail> \ template <typename... Tail> \
struct InferMetaFnCallHelper<attr_type, Tail...> { \ struct InferMetaFnCallHelper<attr_type, Tail...> { \
template <int in_idx, int attr_idx, int out_idx, typename... PreviousArgs> \ template <int in_idx, int attr_idx, int out_idx, typename... PreviousArgs> \
...@@ -175,24 +175,24 @@ struct InferMetaFnImpl<Return (*)(Args...), infer_meta_fn> { ...@@ -175,24 +175,24 @@ struct InferMetaFnImpl<Return (*)(Args...), infer_meta_fn> {
}; };
// TODO(chenweihang): support other attr type later // TODO(chenweihang): support other attr type later
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(bool); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(bool);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(int); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(int);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(int64_t); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(int64_t);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(float); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(float);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::string&); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::string&);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::vector<bool>&); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::vector<bool>&);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::vector<int>&); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::vector<int>&);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE( PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(
const std::vector<int64_t>&); const std::vector<int64_t>&);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::vector<float>&); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::vector<float>&);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::vector<double>&); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const std::vector<double>&);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE( PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(
const std::vector<std::string>&); const std::vector<std::string>&);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(DataType); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(DataType);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(Backend); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(Backend);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(DataLayout); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(DataLayout);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const Scalar&); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const Scalar&);
PT_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const ScalarArray&); PD_SPECIALIZE_InferMetaFnCallHelper_FOR_ATTRIBUTE(const ScalarArray&);
// TODO(chenweihang): support vector<MetaTensor> input later // TODO(chenweihang): support vector<MetaTensor> input later
...@@ -304,11 +304,11 @@ struct InferMetaFnRegistrar { ...@@ -304,11 +304,11 @@ struct InferMetaFnRegistrar {
}; };
#define PD_REGISTER_INFER_META_FN(kernel_name_prefix, variadic_infer_meta_fn) \ #define PD_REGISTER_INFER_META_FN(kernel_name_prefix, variadic_infer_meta_fn) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PD_STATIC_ASSERT_GLOBAL_NAMESPACE( \
PD_REGISTER_infer_meta_fn_ns_check_##kernel_name_prefix, \ PD_REGISTER_infer_meta_fn_ns_check_##kernel_name_prefix, \
"PD_REGISTER_INFER_META_FN must be called in global namespace."); \ "PD_REGISTER_INFER_META_FN must be called in global namespace."); \
static const ::phi::InferMetaFnRegistrar \ static const ::phi::InferMetaFnRegistrar \
__registrar_arg_map_fn_for_##kernel_name_prefix( \ __registrar_arg_map_fn_for_##kernel_name_prefix( \
#kernel_name_prefix, PT_INFER_META(variadic_infer_meta_fn)) #kernel_name_prefix, PD_INFER_META(variadic_infer_meta_fn))
} // namespace phi } // namespace phi
paddle/phi/core/kernel_registry.h
浏览文件 @ d30d85da
...@@ -228,13 +228,13 @@ struct KernelRegistrar { ...@@ -228,13 +228,13 @@ struct KernelRegistrar {
* http://connect.microsoft.com/VisualStudio/feedback/details/380090/variadic-macro-replacement * http://connect.microsoft.com/VisualStudio/feedback/details/380090/variadic-macro-replacement
* http://cplusplus.co.il/2010/07/17/variadic-macro-to-count-number-of-arguments/#comment-644 * http://cplusplus.co.il/2010/07/17/variadic-macro-to-count-number-of-arguments/#comment-644
*/ */
#define PT_NARGS(...) _PT_NARGS((__VA_ARGS__, _PT_RESQ_N())) #define PD_NARGS(...) _PD_NARGS((__VA_ARGS__, _PD_RESQ_N()))
#define _PT_NARGS(...) _PT_ARG_N(__VA_ARGS__) #define _PD_NARGS(...) _PD_ARG_N(__VA_ARGS__)
#define _PT_ARG_N_EXPAND( \ #define _PD_ARG_N_EXPAND( \
_1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, N, ...) \ _1, _2, _3, _4, _5, _6, _7, _8, _9, _10, _11, _12, _13, _14, _15, N, ...) \
N N
#define _PT_ARG_N(args) _PT_ARG_N_EXPAND args #define _PD_ARG_N(args) _PD_ARG_N_EXPAND args
#define _PT_RESQ_N() 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0 #define _PD_RESQ_N() 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0
/** PD_REGISTER_KERNEL /** PD_REGISTER_KERNEL
* *
...@@ -256,10 +256,10 @@ struct KernelRegistrar { ...@@ -256,10 +256,10 @@ struct KernelRegistrar {
#define _PD_REGISTER_KERNEL( \ #define _PD_REGISTER_KERNEL( \
reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \ reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PD_STATIC_ASSERT_GLOBAL_NAMESPACE( \
PD_REGISTER_tp_kernel_ns_check_##kernel_name##_##backend##_##layout, \ PD_REGISTER_tp_kernel_ns_check_##kernel_name##_##backend##_##layout, \
"PD_REGISTER_KERNEL must be called in global namespace."); \ "PD_REGISTER_KERNEL must be called in global namespace."); \
PT_EXPAND(_PD_REGISTER_2TA_KERNEL(reg_type, \ PD_EXPAND(_PD_REGISTER_2TA_KERNEL(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -270,19 +270,19 @@ struct KernelRegistrar { ...@@ -270,19 +270,19 @@ struct KernelRegistrar {
#ifndef _WIN32 #ifndef _WIN32
#define _PD_REGISTER_2TA_KERNEL( \ #define _PD_REGISTER_2TA_KERNEL( \
reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \ reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \
PT_KERNEL_INSTANTIATION(meta_kernel_fn, backend, context, __VA_ARGS__); \ PD_KERNEL_INSTANTIATION(meta_kernel_fn, backend, context, __VA_ARGS__); \
static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ static void __PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \ const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \
PT_KERNEL_REGISTRAR_INIT( \ PD_KERNEL_REGISTRAR_INIT( \
reg_type, \ reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
&__PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout, \ &__PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__); \ __VA_ARGS__); \
void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ void __PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel) const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel)
#else #else
/** /**
...@@ -299,119 +299,119 @@ struct KernelRegistrar { ...@@ -299,119 +299,119 @@ struct KernelRegistrar {
*/ */
#define _PD_REGISTER_2TA_KERNEL( \ #define _PD_REGISTER_2TA_KERNEL( \
reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \ reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \
static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ static void __PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \ const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \
PT_EXPAND(PT_KERNEL_REGISTRAR_INIT( \ PD_EXPAND(PD_KERNEL_REGISTRAR_INIT( \
reg_type, \ reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
&__PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout, \ &__PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)); \ __VA_ARGS__)); \
void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ void __PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel) const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel)
#endif #endif
#define PT_KERNEL_INSTANTIATION(meta_kernel_fn, backend, context, ...) \ #define PD_KERNEL_INSTANTIATION(meta_kernel_fn, backend, context, ...) \
_PT_KERNEL_INSTANTIATION( \ _PD_KERNEL_INSTANTIATION( \
PT_NARGS(__VA_ARGS__), meta_kernel_fn, backend, context, __VA_ARGS__) PD_NARGS(__VA_ARGS__), meta_kernel_fn, backend, context, __VA_ARGS__)
#define _PT_KERNEL_INSTANTIATION(N, meta_kernel_fn, backend, context, ...) \ #define _PD_KERNEL_INSTANTIATION(N, meta_kernel_fn, backend, context, ...) \
PT_CONCATENATE(_PT_KERNEL_INSTANTIATION_, N) \ PD_CONCATENATE(_PD_KERNEL_INSTANTIATION_, N) \
(meta_kernel_fn, backend, context, __VA_ARGS__) (meta_kernel_fn, backend, context, __VA_ARGS__)
#define _PT_KERNEL_INSTANTIATION_1( \ #define _PD_KERNEL_INSTANTIATION_1( \
meta_kernel_fn, backend, context, cpp_dtype) \ meta_kernel_fn, backend, context, cpp_dtype) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context> meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>
#define _PT_KERNEL_INSTANTIATION_2( \ #define _PD_KERNEL_INSTANTIATION_2( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_1( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_1( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_3( \ #define _PD_KERNEL_INSTANTIATION_3( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_2( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_2( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_4( \ #define _PD_KERNEL_INSTANTIATION_4( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_3( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_3( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_5( \ #define _PD_KERNEL_INSTANTIATION_5( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_4( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_4( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_6( \ #define _PD_KERNEL_INSTANTIATION_6( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_5( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_5( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_7( \ #define _PD_KERNEL_INSTANTIATION_7( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_6( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_6( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_8( \ #define _PD_KERNEL_INSTANTIATION_8( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_7( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_7( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_9( \ #define _PD_KERNEL_INSTANTIATION_9( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_8( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_8( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_10( \ #define _PD_KERNEL_INSTANTIATION_10( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_9( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_9( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_11( \ #define _PD_KERNEL_INSTANTIATION_11( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_10( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_10( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_12( \ #define _PD_KERNEL_INSTANTIATION_12( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_11( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_11( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_13( \ #define _PD_KERNEL_INSTANTIATION_13( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_12( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_12( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_14( \ #define _PD_KERNEL_INSTANTIATION_14( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_13( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_13( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define _PT_KERNEL_INSTANTIATION_15( \ #define _PD_KERNEL_INSTANTIATION_15( \
meta_kernel_fn, backend, context, cpp_dtype, ...) \ meta_kernel_fn, backend, context, cpp_dtype, ...) \
template decltype( \ template decltype( \
meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \ meta_kernel_fn<cpp_dtype, context>) meta_kernel_fn<cpp_dtype, context>; \
PT_EXPAND(_PT_KERNEL_INSTANTIATION_14( \ PD_EXPAND(_PD_KERNEL_INSTANTIATION_14( \
meta_kernel_fn, backend, context, __VA_ARGS__)) meta_kernel_fn, backend, context, __VA_ARGS__))
#define PT_KERNEL_REGISTRAR_INIT(reg_type, \ #define PD_KERNEL_REGISTRAR_INIT(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -419,7 +419,7 @@ struct KernelRegistrar { ...@@ -419,7 +419,7 @@ struct KernelRegistrar {
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
...) \ ...) \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT(PT_NARGS(__VA_ARGS__), \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT(PD_NARGS(__VA_ARGS__), \
reg_type, \ reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
...@@ -433,7 +433,7 @@ struct KernelRegistrar { ...@@ -433,7 +433,7 @@ struct KernelRegistrar {
/* The =pre-commit always treats this macro into the wrong format, /* The =pre-commit always treats this macro into the wrong format,
and multi-line macros cannot be skipped with NOLINT.*/ and multi-line macros cannot be skipped with NOLINT.*/
#define _PT_KERNEL_REGISTRAR_INIT(N, \ #define _PD_KERNEL_REGISTRAR_INIT(N, \
reg_type, \ reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
...@@ -442,20 +442,20 @@ struct KernelRegistrar { ...@@ -442,20 +442,20 @@ struct KernelRegistrar {
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
...) \ ...) \
PT_EXPAND(PT_CONCATENATE(_PT_KERNEL_REGISTRAR_INIT_, N) ( \ PD_EXPAND(PD_CONCATENATE(_PD_KERNEL_REGISTRAR_INIT_, N) ( \
reg_type, \ reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
// clang-format on // clang-format on
#define _PT_KERNEL_REGISTRAR_INIT_1(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_1(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -464,7 +464,7 @@ struct KernelRegistrar { ...@@ -464,7 +464,7 @@ struct KernelRegistrar {
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype) \ cpp_dtype) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -474,10 +474,10 @@ struct KernelRegistrar { ...@@ -474,10 +474,10 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout() { return 0; } int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout() { return 0; }
#define _PT_KERNEL_REGISTRAR_INIT_2(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_2(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -487,7 +487,7 @@ struct KernelRegistrar { ...@@ -487,7 +487,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -497,18 +497,18 @@ struct KernelRegistrar { ...@@ -497,18 +497,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_1(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_1(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_3(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_3(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -518,7 +518,7 @@ struct KernelRegistrar { ...@@ -518,7 +518,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -528,18 +528,18 @@ struct KernelRegistrar { ...@@ -528,18 +528,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_2(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_2(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_4(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_4(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -549,7 +549,7 @@ struct KernelRegistrar { ...@@ -549,7 +549,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -559,18 +559,18 @@ struct KernelRegistrar { ...@@ -559,18 +559,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_3(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_3(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_5(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_5(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -580,7 +580,7 @@ struct KernelRegistrar { ...@@ -580,7 +580,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -590,18 +590,18 @@ struct KernelRegistrar { ...@@ -590,18 +590,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_4(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_4(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_6(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_6(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -611,7 +611,7 @@ struct KernelRegistrar { ...@@ -611,7 +611,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -621,18 +621,18 @@ struct KernelRegistrar { ...@@ -621,18 +621,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_5(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_5(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_7(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_7(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -642,7 +642,7 @@ struct KernelRegistrar { ...@@ -642,7 +642,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -652,18 +652,18 @@ struct KernelRegistrar { ...@@ -652,18 +652,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_6(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_6(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_8(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_8(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -673,7 +673,7 @@ struct KernelRegistrar { ...@@ -673,7 +673,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -683,18 +683,18 @@ struct KernelRegistrar { ...@@ -683,18 +683,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_7(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_7(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_9(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_9(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -704,7 +704,7 @@ struct KernelRegistrar { ...@@ -704,7 +704,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -714,18 +714,18 @@ struct KernelRegistrar { ...@@ -714,18 +714,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_8(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_8(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_10(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_10(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -735,7 +735,7 @@ struct KernelRegistrar { ...@@ -735,7 +735,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -745,18 +745,18 @@ struct KernelRegistrar { ...@@ -745,18 +745,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_9(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_9(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_11(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_11(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -766,7 +766,7 @@ struct KernelRegistrar { ...@@ -766,7 +766,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -776,18 +776,18 @@ struct KernelRegistrar { ...@@ -776,18 +776,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_10(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_10(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_12(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_12(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -797,7 +797,7 @@ struct KernelRegistrar { ...@@ -797,7 +797,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -807,18 +807,18 @@ struct KernelRegistrar { ...@@ -807,18 +807,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_11(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_11(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_13(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_13(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -828,7 +828,7 @@ struct KernelRegistrar { ...@@ -828,7 +828,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -838,18 +838,18 @@ struct KernelRegistrar { ...@@ -838,18 +838,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_12(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_12(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_14(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_14(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -859,7 +859,7 @@ struct KernelRegistrar { ...@@ -859,7 +859,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -869,18 +869,18 @@ struct KernelRegistrar { ...@@ -869,18 +869,18 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_13(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_13(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
#define _PT_KERNEL_REGISTRAR_INIT_15(reg_type, \ #define _PD_KERNEL_REGISTRAR_INIT_15(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -890,7 +890,7 @@ struct KernelRegistrar { ...@@ -890,7 +890,7 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
cpp_dtype, \ cpp_dtype, \
...) \ ...) \
static const ::phi::KernelRegistrar PT_CONCATENATE( \ static const ::phi::KernelRegistrar PD_CONCATENATE( \
__reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout##_, registrar_id)( \
reg_type, \ reg_type, \
#kernel_name, \ #kernel_name, \
...@@ -900,14 +900,14 @@ struct KernelRegistrar { ...@@ -900,14 +900,14 @@ struct KernelRegistrar {
::phi::KernelArgsParseFunctor<decltype( \ ::phi::KernelArgsParseFunctor<decltype( \
&meta_kernel_fn<cpp_dtype, context>)>::Parse, \ &meta_kernel_fn<cpp_dtype, context>)>::Parse, \
args_def_fn, \ args_def_fn, \
PT_KERNEL(meta_kernel_fn<cpp_dtype, context>), \ PHI_KERNEL(meta_kernel_fn<cpp_dtype, context>), \
PT_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \ PHI_VARIADIC_KERNEL(meta_kernel_fn<cpp_dtype, context>)); \
PT_EXPAND(_PT_KERNEL_REGISTRAR_INIT_14(reg_type, \ PD_EXPAND(_PD_KERNEL_REGISTRAR_INIT_14(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
layout, \ layout, \
PT_ID, \ PD_ID, \
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
...@@ -924,7 +924,7 @@ struct KernelRegistrar { ...@@ -924,7 +924,7 @@ struct KernelRegistrar {
#define _PD_REGISTER_GENERAL_KERNEL( \ #define _PD_REGISTER_GENERAL_KERNEL( \
reg_type, kernel_name, backend, layout, kernel_fn, dtype) \ reg_type, kernel_name, backend, layout, kernel_fn, dtype) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PD_STATIC_ASSERT_GLOBAL_NAMESPACE( \
PD_REGISTER_no_t_kernel_ns_check_##kernel_name##_##backend##_##layout, \ PD_REGISTER_no_t_kernel_ns_check_##kernel_name##_##backend##_##layout, \
"PD_REGISTER_NO_TEMPLATE_KERNEL must be called in global namespace."); \ "PD_REGISTER_NO_TEMPLATE_KERNEL must be called in global namespace."); \
__PD_REGISTER_GENERAL_KERNEL( \ __PD_REGISTER_GENERAL_KERNEL( \
...@@ -934,7 +934,7 @@ struct KernelRegistrar { ...@@ -934,7 +934,7 @@ struct KernelRegistrar {
#define __PD_REGISTER_GENERAL_KERNEL( \ #define __PD_REGISTER_GENERAL_KERNEL( \
reg_type, kernel_name, backend, layout, kernel_fn, dtype) \ reg_type, kernel_name, backend, layout, kernel_fn, dtype) \
template decltype(kernel_fn) kernel_fn; \ template decltype(kernel_fn) kernel_fn; \
static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ static void __PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \ const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \
static const ::phi::KernelRegistrar \ static const ::phi::KernelRegistrar \
__reg_pt_kernel_##kernel_name##_##backend##_##layout( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout( \
...@@ -943,18 +943,18 @@ struct KernelRegistrar { ...@@ -943,18 +943,18 @@ struct KernelRegistrar {
#backend, \ #backend, \
DATALAYOUT(layout), \ DATALAYOUT(layout), \
::phi::KernelArgsParseFunctor<decltype(&kernel_fn)>::Parse, \ ::phi::KernelArgsParseFunctor<decltype(&kernel_fn)>::Parse, \
&__PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout, \ &__PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout, \
PT_KERNEL(kernel_fn), \ PHI_KERNEL(kernel_fn), \
PT_VARIADIC_KERNEL(kernel_fn)); \ PHI_VARIADIC_KERNEL(kernel_fn)); \
int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout() { \ int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout() { \
return 0; \ return 0; \
} \ } \
void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ void __PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel) const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel)
#else #else
#define __PD_REGISTER_GENERAL_KERNEL( \ #define __PD_REGISTER_GENERAL_KERNEL( \
reg_type, kernel_name, backend, layout, kernel_fn, dtype) \ reg_type, kernel_name, backend, layout, kernel_fn, dtype) \
static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ static void __PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \ const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \
static const ::phi::KernelRegistrar \ static const ::phi::KernelRegistrar \
__reg_pt_kernel_##kernel_name##_##backend##_##layout( \ __reg_pt_kernel_##kernel_name##_##backend##_##layout( \
...@@ -963,13 +963,13 @@ struct KernelRegistrar { ...@@ -963,13 +963,13 @@ struct KernelRegistrar {
#backend, \ #backend, \
DATALAYOUT(layout), \ DATALAYOUT(layout), \
::phi::KernelArgsParseFunctor<decltype(&kernel_fn)>::Parse, \ ::phi::KernelArgsParseFunctor<decltype(&kernel_fn)>::Parse, \
&__PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout, \ &__PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout, \
PT_KERNEL(kernel_fn), \ PHI_KERNEL(kernel_fn), \
PT_VARIADIC_KERNEL(kernel_fn)); \ PHI_VARIADIC_KERNEL(kernel_fn)); \
int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout() { \ int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout() { \
return 0; \ return 0; \
} \ } \
void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ void __PD_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel) const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel)
#endif #endif
...@@ -979,7 +979,7 @@ struct KernelRegistrar { ...@@ -979,7 +979,7 @@ struct KernelRegistrar {
* to avoid being removed by linker * to avoid being removed by linker
*/ */
#define PD_DECLARE_KERNEL(kernel_name, backend, layout) \ #define PD_DECLARE_KERNEL(kernel_name, backend, layout) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PD_STATIC_ASSERT_GLOBAL_NAMESPACE( \
PD_DECLARE_tp_kernel_ns_check_##kernel_name##_##backend##_##layout, \ PD_DECLARE_tp_kernel_ns_check_##kernel_name##_##backend##_##layout, \
"PD_DECLARE_KERNEL must be called in global namespace."); \ "PD_DECLARE_KERNEL must be called in global namespace."); \
extern int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout(); \ extern int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout(); \
......
paddle/phi/core/kernel_utils.h
浏览文件 @ d30d85da
...@@ -30,14 +30,15 @@ ...@@ -30,14 +30,15 @@
namespace phi { namespace phi {
#define PT_KERNEL(...) \ // PD_KERNEL has been used by custom op api
#define PHI_KERNEL(...) \
::phi::KernelImpl<decltype(&__VA_ARGS__), &__VA_ARGS__>::Compute ::phi::KernelImpl<decltype(&__VA_ARGS__), &__VA_ARGS__>::Compute
#define PT_VARIADIC_KERNEL(...) \ #define PHI_VARIADIC_KERNEL(...) \
reinterpret_cast<void*>(&::phi::KernelImpl<decltype(&__VA_ARGS__), \ reinterpret_cast<void*>(&::phi::KernelImpl<decltype(&__VA_ARGS__), \
&__VA_ARGS__>::VariadicCompute) &__VA_ARGS__>::VariadicCompute)
#define PT_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(dev_ctx) \ #define PD_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(dev_ctx) \
template <typename... Tail> \ template <typename... Tail> \
struct KernelCallHelper<const dev_ctx&, Tail...> { \ struct KernelCallHelper<const dev_ctx&, Tail...> { \
template <int dev_ctx_idx, \ template <int dev_ctx_idx, \
...@@ -60,7 +61,7 @@ namespace phi { ...@@ -60,7 +61,7 @@ namespace phi {
} \ } \
} }
#define PT_SPECIALIZE_KernelCallHelper_FOR_INPUT(tensor_type) \ #define PD_SPECIALIZE_KernelCallHelper_FOR_INPUT(tensor_type) \
template <typename... Tail> \ template <typename... Tail> \
struct KernelCallHelper<const tensor_type&, Tail...> { \ struct KernelCallHelper<const tensor_type&, Tail...> { \
template <int dev_ctx_idx, \ template <int dev_ctx_idx, \
...@@ -81,7 +82,7 @@ namespace phi { ...@@ -81,7 +82,7 @@ namespace phi {
} \ } \
} }
#define PT_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(tensor_type) \ #define PD_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(tensor_type) \
template <typename... Tail> \ template <typename... Tail> \
struct KernelCallHelper<paddle::optional<const tensor_type&>, Tail...> { \ struct KernelCallHelper<paddle::optional<const tensor_type&>, Tail...> { \
template <int dev_ctx_idx, \ template <int dev_ctx_idx, \
...@@ -102,7 +103,7 @@ namespace phi { ...@@ -102,7 +103,7 @@ namespace phi {
} \ } \
} }
#define PT_SPECIALIZE_KernelCallHelper_FOR_MULTI_INPUT(tensor_type) \ #define PD_SPECIALIZE_KernelCallHelper_FOR_MULTI_INPUT(tensor_type) \
template <typename... Tail> \ template <typename... Tail> \
struct KernelCallHelper<const std::vector<const tensor_type*>&, Tail...> { \ struct KernelCallHelper<const std::vector<const tensor_type*>&, Tail...> { \
template <int dev_ctx_idx, \ template <int dev_ctx_idx, \
...@@ -124,7 +125,7 @@ namespace phi { ...@@ -124,7 +125,7 @@ namespace phi {
} \ } \
} }
#define PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(attr_type) \ #define PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(attr_type) \
template <typename... Tail> \ template <typename... Tail> \
struct KernelCallHelper<attr_type, Tail...> { \ struct KernelCallHelper<attr_type, Tail...> { \
template <int dev_ctx_idx, \ template <int dev_ctx_idx, \
...@@ -142,7 +143,7 @@ namespace phi { ...@@ -142,7 +143,7 @@ namespace phi {
} \ } \
} }
#define PT_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(tensor_type) \ #define PD_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(tensor_type) \
template <typename... Tail> \ template <typename... Tail> \
struct KernelCallHelper<tensor_type*, Tail...> { \ struct KernelCallHelper<tensor_type*, Tail...> { \
template <int dev_ctx_idx, \ template <int dev_ctx_idx, \
...@@ -159,7 +160,7 @@ namespace phi { ...@@ -159,7 +160,7 @@ namespace phi {
} \ } \
} }
#define PT_SPECIALIZE_KernelCallHelper_FOR_MULTI_OUTPUT(tensor_type) \ #define PD_SPECIALIZE_KernelCallHelper_FOR_MULTI_OUTPUT(tensor_type) \
template <typename... Tail> \ template <typename... Tail> \
struct KernelCallHelper<std::vector<tensor_type*>, Tail...> { \ struct KernelCallHelper<std::vector<tensor_type*>, Tail...> { \
template <int dev_ctx_idx, \ template <int dev_ctx_idx, \
...@@ -204,65 +205,65 @@ struct KernelImpl<Return (*)(DevCtx, Args...), kernel_fn> { ...@@ -204,65 +205,65 @@ struct KernelImpl<Return (*)(DevCtx, Args...), kernel_fn> {
/* DeviceContext Helpers */ /* DeviceContext Helpers */
PT_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(CPUContext); PD_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(CPUContext);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(GPUContext); PD_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(GPUContext);
#endif #endif
#ifdef PADDLE_WITH_XPU #ifdef PADDLE_WITH_XPU
PT_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(XPUContext); PD_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(XPUContext);
#endif #endif
#ifdef PADDLE_WITH_CUSTOM_DEVICE #ifdef PADDLE_WITH_CUSTOM_DEVICE
PT_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(CustomContext); PD_SPECIALIZE_KernelCallHelper_FOR_DEVICE_CONTEXT(CustomContext);
#endif #endif
/* Input Helpers */ /* Input Helpers */
PT_SPECIALIZE_KernelCallHelper_FOR_INPUT(DenseTensor); PD_SPECIALIZE_KernelCallHelper_FOR_INPUT(DenseTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(DenseTensor); PD_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(DenseTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(SelectedRows); PD_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(SelectedRows);
PT_SPECIALIZE_KernelCallHelper_FOR_MULTI_INPUT(DenseTensor); PD_SPECIALIZE_KernelCallHelper_FOR_MULTI_INPUT(DenseTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_INPUT(SelectedRows); PD_SPECIALIZE_KernelCallHelper_FOR_INPUT(SelectedRows);
PT_SPECIALIZE_KernelCallHelper_FOR_INPUT(SparseCooTensor); PD_SPECIALIZE_KernelCallHelper_FOR_INPUT(SparseCooTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(SparseCooTensor); PD_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(SparseCooTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_MULTI_INPUT(SparseCooTensor); PD_SPECIALIZE_KernelCallHelper_FOR_MULTI_INPUT(SparseCooTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_INPUT(SparseCsrTensor); PD_SPECIALIZE_KernelCallHelper_FOR_INPUT(SparseCsrTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(SparseCsrTensor); PD_SPECIALIZE_KernelCallHelper_FOR_OPTIONAL_INPUT(SparseCsrTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_MULTI_INPUT(SparseCsrTensor); PD_SPECIALIZE_KernelCallHelper_FOR_MULTI_INPUT(SparseCsrTensor);
/* Attribute Helpers */ /* Attribute Helpers */
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(bool); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(bool);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(float); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(float);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(double); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(double);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(int); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(int);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(int64_t); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(int64_t);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(phi::dtype::float16); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(phi::dtype::float16);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const Scalar&); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const Scalar&);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(DataType); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(DataType);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(DataLayout); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(DataLayout);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(Place); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(Place);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<int64_t>&); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<int64_t>&);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const ScalarArray&); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const ScalarArray&);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<int>&); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<int>&);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::string&); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::string&);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<bool>&); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<bool>&);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<float>&); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<float>&);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<double>&); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<double>&);
PT_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<std::string>&); PD_SPECIALIZE_KernelCallHelper_FOR_ATTRIBUTE(const std::vector<std::string>&);
/* Output Helpers */ /* Output Helpers */
PT_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(DenseTensor); PD_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(DenseTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_MULTI_OUTPUT(DenseTensor); PD_SPECIALIZE_KernelCallHelper_FOR_MULTI_OUTPUT(DenseTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(SelectedRows); PD_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(SelectedRows);
PT_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(SparseCooTensor); PD_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(SparseCooTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_MULTI_OUTPUT(SparseCooTensor); PD_SPECIALIZE_KernelCallHelper_FOR_MULTI_OUTPUT(SparseCooTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(SparseCsrTensor); PD_SPECIALIZE_KernelCallHelper_FOR_OUTPUT(SparseCsrTensor);
PT_SPECIALIZE_KernelCallHelper_FOR_MULTI_OUTPUT(SparseCsrTensor); PD_SPECIALIZE_KernelCallHelper_FOR_MULTI_OUTPUT(SparseCsrTensor);
/* End case */ /* End case */
template <typename T> template <typename T>
......
paddle/phi/core/macros.h
浏览文件 @ d30d85da
...@@ -26,19 +26,19 @@ namespace phi { ...@@ -26,19 +26,19 @@ namespace phi {
classname& operator=(classname&&) = delete classname& operator=(classname&&) = delete
#endif #endif
#define PT_STATIC_ASSERT_GLOBAL_NAMESPACE(uniq_name, msg) \ #define PD_STATIC_ASSERT_GLOBAL_NAMESPACE(uniq_name, msg) \
_PT_STATIC_ASSERT_GLOBAL_NAMESPACE(uniq_name, msg) _PD_STATIC_ASSERT_GLOBAL_NAMESPACE(uniq_name, msg)
#define _PT_STATIC_ASSERT_GLOBAL_NAMESPACE(uniq_name, msg) \ #define _PD_STATIC_ASSERT_GLOBAL_NAMESPACE(uniq_name, msg) \
struct __test_global_namespace_##uniq_name##__ {}; \ struct __test_global_namespace_##uniq_name##__ {}; \
static_assert(std::is_same<::__test_global_namespace_##uniq_name##__, \ static_assert(std::is_same<::__test_global_namespace_##uniq_name##__, \
__test_global_namespace_##uniq_name##__>::value, \ __test_global_namespace_##uniq_name##__>::value, \
msg) msg)
#ifdef __COUNTER__ #ifdef __COUNTER__
#define PT_ID __COUNTER__ #define PD_ID __COUNTER__
#else #else
#define PT_ID __LINE__ #define PD_ID __LINE__
#endif #endif
#if defined(_WIN32) #if defined(_WIN32)
...@@ -48,9 +48,9 @@ namespace phi { ...@@ -48,9 +48,9 @@ namespace phi {
#define UNUSED __attribute__((unused)) #define UNUSED __attribute__((unused))
#endif #endif
#define PT_CONCATENATE(arg1, arg2) PT_CONCATENATE1(arg1, arg2) #define PD_CONCATENATE(arg1, arg2) PD_CONCATENATE1(arg1, arg2)
#define PT_CONCATENATE1(arg1, arg2) PT_CONCATENATE2(arg1, arg2) #define PD_CONCATENATE1(arg1, arg2) PD_CONCATENATE2(arg1, arg2)
#define PT_CONCATENATE2(arg1, arg2) arg1##arg2 #define PD_CONCATENATE2(arg1, arg2) arg1##arg2
#define PT_EXPAND(x) x #define PD_EXPAND(x) x
} // namespace phi } // namespace phi
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或