[PIR][Inference] add layout_transformation_interface for later transfer_layout_pass

[kangguangli]

PR Category

Execute Infrastructure

PR Types

Devs

Description

This PR adds LayoutTransformationInterface, which is used for transfer_layout_pass to determine which layout allows an operation to achieve the best performance.

LayoutTransformationInterface has four member functions:

1. PreferLayout, receives an operation as input and judges the layout that this operation will perform best by its attributes、inputs or operation kind and so on. This can be overloaded. If not, its result is AllLayout which means it can be modified freely.
2. RewriteByLayout, receives an operation and a new layout as input, exectutes after we choose a new layout and rewrite the operation's attributes. This must be overloaded, otherwise it throws an error.
3. RelevantInputs, receives an operation as input, returns the part of op's inputs that affects its kernel layout. This can be overloaded. If not, its result is all inputs of the op.
4. RelevantOutputs, receives an operation as input, returns the part of op's outputs that are affected by its kernel layout. This can be overloaded. If not, its result is all outputs of the op.

To register this inferface for an op after this PR, you will need to modify 3 files:

1. ops.yaml, to add following code: interfaces : paddle::dialect::LayoutTransformationInterface
2. paddle/fluid/pir/dialect/operator/interface/layout_transformation.hpp, to add the template declaration. Note that you need a forward declaration to avoid circular reference and only declare those members that you need to overload.

class FusedConv2dAddActOp;
template <> common::DataLayout PreferLayoutImpl<FusedConv2dAddActOp>(pir::Operation*);
extern template common::DataLayout PreferLayoutImpl<FusedConv2dAddActOp>(pir::Operation*);
template <>void RewriteByLayoutImpl<FusedConv2dAddActOp>(pir::Operation*,  common::DataLayout);
extern template void RewriteByLayoutImpl<FusedConv2dAddActOp>(pir::Operation*, common::DataLayout);

3. paddle/fluid/pir/dialect/operator/interface/layout_transformation.cc, to add the specialized template implementation

Others

Pcard-67164

[paddle-bot]

你的PR提交成功，感谢你对开源项目的贡献!
请关注后续CI自动化测试结果，详情请参考Paddle-CI手册。
Your PR has been submitted. Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

[kangguangli]

暂时留空，后面写pass的时候补上