Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[PIR] Support translate IfOp #57342

Merged
merged 16 commits into from
Sep 16, 2023
Merged

[PIR] Support translate IfOp #57342

Show file tree
Hide file tree
Changes from 14 commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
3af7287
add code
zhangbo9674 Sep 14, 2023
2e2fee8
refine code
zhangbo9674 Sep 15, 2023
983e2cc
add code
zhangbo9674 Sep 15, 2023
73ea47d
add code
zhangbo9674 Sep 15, 2023
d85d5ac
fix bug
zhangbo9674 Sep 15, 2023
bf18c44
add code
zhangbo9674 Sep 15, 2023
12c3451
add code
zhangbo9674 Sep 15, 2023
022248d
add code
zhangbo9674 Sep 15, 2023
9faee3b
add code
zhangbo9674 Sep 15, 2023
17355d6
add code
zhangbo9674 Sep 15, 2023
1ee8d79
add code
zhangbo9674 Sep 15, 2023
05286f5
addd code
zhangbo9674 Sep 15, 2023
9107005
refine code
zhangbo9674 Sep 15, 2023
d38b297
refine code
zhangbo9674 Sep 15, 2023
b883667
fix conflict
zhangbo9674 Sep 15, 2023
86e86ac
fix conflict
zhangbo9674 Sep 15, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
122 changes: 61 additions & 61 deletions paddle/fluid/ir_adaptor/translator/op_translator.cc
View file
Open in desktop

Large diffs are not rendered by default.

12 changes: 6 additions & 6 deletions paddle/fluid/ir_adaptor/translator/op_translator.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -54,15 +54,15 @@ struct OpTranscriber {
const OpDesc&,
const std::string&,
const OpInputInfo&,
pir::Program*)>;
pir::Block*)>;
using AttributeHandlerFn = std::function<pir::Attribute(
pir::IrContext*, const OpDesc&, const OpAttributeInfo&)>;

public:
virtual pir::Operation* operator()(pir::IrContext* ctx,
TranslationContext* param_map,
const OpDesc& op_desc,
pir::Program* program);
pir::Block* block);

public:
virtual pir::OpInfo LoopkUpOpInfo(pir::IrContext* ctx, const OpDesc& op_desc);
Expand All @@ -72,7 +72,7 @@ struct OpTranscriber {
const OpDesc& op_desc,
const std::string& normalized_op_name,
const OpInputInfoList& input_infos,
pir::Program* program);
pir::Block* block);
virtual std::tuple<OpOutputTypeList, OpOutputMapping> GenerateOperationOutput(
pir::IrContext* ctx,
const OpDesc& op_desc,
Expand All @@ -86,7 +86,7 @@ struct OpTranscriber {
const OpAttributeInfoList& op_attr_infos,
const OpDesc& op_desc);
virtual pir::OpResult GetAttributeAsInput(pir::IrContext* ctx,
pir::Program* program,
pir::Block* block,
const OpDesc& op_desc,
const OpInputInfo& input_info);

Expand All @@ -109,7 +109,7 @@ struct OpTranscriber {
TranslationContext* param_map,
const OpDesc& op_desc,
const OpInputInfoList& input_infos,
pir::Program* program);
pir::Block* block);
};

class OpTranslator {
Expand All @@ -119,7 +119,7 @@ class OpTranslator {
using BlockDesc = paddle::framework::BlockDesc;
using VarDesc = paddle::framework::VarDesc;
using OpTranslateFn = std::function<pir::Operation*(
pir::IrContext*, TranslationContext*, const OpDesc&, pir::Program*)>;
pir::IrContext*, TranslationContext*, const OpDesc&, pir::Block*)>;

private:
OpTranslator(); // Disallow instantiation outside of the class.
Expand Down
Loading