[PTen] Move GetExpectedPtenKernelArgs function into pten for infrt

[chenwhql]

PR types

Function optimization

PR changes

Others

Describe

[PTen] Move GetExpectedPtenKernelArgs function into pten for infrt

infrt在执行时，会直接载入program解析，并根据program的信息去选择kernel，但不会依赖paddle的op体系，因此，用于Op参数和kernel参数的映射函数GetExpectedPtenKernelArgs需要作为pten的组件，可以由框架灵活使用，并且不反向依赖fluid的实现

1. 对fluid的适配，见代码中修改的示例

• TODO：适配上，可以直接放到执行体系中，而不是在每个op中都重写GetExpectedPtenKernelArgs方法，鉴于一次性修改涉及文件较多，会在后续PR逐步修改

2. 对infrt的适配，可以通过继承创建合适的ArgumentMappingContext，从而在infrt中调用相应匹配函数

// 伪代码 class ProtoArgumentMappingContext : public pten::ArgumentMappingContext { public: ProtoArgumentMappingContext(proto::OpProto* op, proto::BlockDesc* block) : op_(op), block_(block) {} bool HasInput(const std::string& name) const override { // simple search for (int i = 0; i < proto_->input_size(); ++i) { auto& in = proto_->inputs()[i]; if (in.name() == name) { return true; } } return false; } bool HasOutput(const std::string& name) const override { // simple search for (int i = 0; i < proto_->output_size(); ++i) { auto& out = proto_->outputs()[i]; if (out.name() == name) { return true; } } return false; } bool HasAttr(const std::string& name) const override { // simple search for (int i = 0; i < proto_->attrs_size(); ++i) { auto& attr = proto_->attrs()[i]; if (attr.name() == name) { return true; } } return false; } size_t InputSize(const std::string& name) const override { return proto_->input_size(); } size_t OutputSize(const std::string& name) const override { return proto_->output_size(); } bool IsDenseTensorInput(const std::string& name) const override { for (int i = 0; i < block_.vars_size(); ++i) { auto& var = block_.vars()[i]; if (var.name() == name) { if (var.type() == proto::VarType::LOD_TENSOR) { return true; } } } // TODO(chenweihang): throw error when cannot found return false; } bool IsSelectedRowsInput(const std::string& name) const override { for (int i = 0; i < block_.vars_size(); ++i) { auto& var = block_.vars()[i]; if (var.name() == name) { if (var.type() == proto::VarType::SELECTED_ROWS) { return true; } } } // TODO(chenweihang): throw error when cannot found return false; } private: proto::OpProto op_*; proto::BlockDesc block_*; }; 

[paddle-bot-old]

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

[zyfncg]

感觉如果这个函数也能注册到一个Map里，原来的Op可能就不用再单独写GetExpectedPtenKernelArgs了

[chenwhql]

是的，下一个PR完成

[zhiqiu]

LGTM