Add FLAGS_low_precision_op_list to get amp list of current module #48843
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
你的PR提交成功，感谢你对开源项目的贡献! |
|
|
需要加一个开关，只在调试模式下才记录低精度算子 -> 开关已加 |
AnnaTrainingG
force-pushed
the
amp_op_list
branch
3 times, most recently
from
64a8e99 to
b8743d3
Compare
zkh2016
previously approved these changes
Dec 20, 2022
zhangting2020
previously approved these changes
Dec 20, 2022
paddle/fluid/pybind/pybind.cc
Outdated
| @@ -2543,6 +2543,10 @@ All parameter, weight, gradient are variables in Paddle. | |||
| m.def("update_autotune_status", | |||
| [] { return phi::autotune::AutoTuneStatus::Instance().Update(); }); | |||
|
|
|||
| m.def("get_low_pricision_op_list", [] { | |||
There was a problem hiding this comment.
pricision -> precision，下面还有几处拼写错误。下个PR改吧
| # amp list elementwise_add, cast | ||
| with paddle.amp.auto_cast(): | ||
| c = a + b | ||
| paddle.fluid.dygraph.amp.auto_cast.low_pricision_op_list() |
There was a problem hiding this comment.
单测可以加上对结果正确性的判断，避免后续将功能改坏
AnnaTrainingG
dismissed stale reviews from zhangting2020 and zkh2016
via
2581df9
AnnaTrainingG
force-pushed
the
amp_op_list
branch
from
67c6aef to
2581df9
Compare
zhangting2020
approved these changes
Dec 21, 2022
zkh2016
approved these changes
Dec 21, 2022
AnnaTrainingG
changed the title
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Others
PR changes
Others
Describe
新增AMP lIst成员，用于记录当前模型中运行的amp算子
添加 export FLAGS_low_precision_op_list=1 环境变量控制低精度OP列表的统计，当 FLAGS_low_precision_op_list=1 时开始统计当前模型中使用amp计算的算子，否则不统计.（说明：当前FLAG在后续开发完正式接口后会删除）
用法
1.开启环境变量：FLAGS_low_precision_op_list=1
2.在模型最后加入： paddle.fluid.dygraph.amp.auto_cast.low_pricision_op_list()
将会打印对应的低精度OP列表