CARVIEW |
Navigation Menu
-
Notifications
You must be signed in to change notification settings - Fork 5.8k
[AutoParallel] Add op_role_guard for PIR #68796
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.
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
[AutoParallel] Add op_role_guard for PIR #68796
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
paddle/pir/include/core/builder.h
Outdated
|
||
// by now the op_role is used by autoparallel for distinguish the op in fw, | ||
// bw, opt region. | ||
int current_op_role_ = -1; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
int current_op_role_ = -1; | |
int op_role_ = -1; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed
paddle/pir/include/core/builder.h
Outdated
@@ -115,7 +115,9 @@ class Builder { | |||
common::errors::PreconditionNotMet("argument of block is nullptr")); | |||
set_insertion_point(block, block->end()); | |||
} | |||
|
|||
/// Set/Get the op_role | |||
void SetCurrentOpRole(int op_role) { current_op_role_ = op_role; } |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
void SetCurrentOpRole(int op_role) { current_op_role_ = op_role; } | |
void set_op_role(int op_role) { op_role_ = op_role; } |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed
paddle/pir/include/core/builder.h
Outdated
|
||
/// Set/Get the op_role | ||
void SetCurrentOpRole(int op_role) { current_op_role_ = op_role; } | ||
int GetCurrentOpRole() const { return current_op_role_; } |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
int GetCurrentOpRole() const { return current_op_role_; } | |
int op_role() const { return op_role_; } |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
fixed
PR Category
Auto Parallel
PR Types
Improvements
Description
Add op_role_guard for PIR to Optimize Graph compilation for Pipeline logic
Reduce the DAG Rewrite Compilation time in pir parallel
Pcard-67164