-
Notifications
You must be signed in to change notification settings - Fork 5.9k
enhance memory optimization transpiler to support user defined skip_opt_set #11372
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
Conversation
|
|
||
|
|
||
| def memory_optimize(input_program, print_log=False, level=0): | ||
| def memory_optimize(input_program, skip_opt_set=None, print_log=False, level=0): |
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.
- Need to add comments for the new arguments.
- Need unit test.
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.
直接修改本地安装环境中对应文件,进行测试,获取的loss值依旧异常
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.
@zchengquan 你是从源码编译安装吗?
不是很确定直接修改本地安装环境中的对应文件,能否使得这个skip生效。
如果是从源码编译安装的话,在修改对应python文件后,请重新执行make install
dzhwinter
left a comment
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.
Try to fix #11320