-
Notifications
You must be signed in to change notification settings - Fork 5.7k
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
fix exponential op #58029
fix exponential op #58029
Changes from 7 commits
7621a5e
f6f3344
546a461
221f686
72b2232
0bf972d
ce68891
8e6aedc
dfe7f18
3cc8c13
1efb4d1
e12c550
30b1c3f
d0d968e
dc48f14
e356fab
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -1439,6 +1439,8 @@ def _check_ir_output(self, place, program, feed_map, fetch_list, outs): | |
check_method = lambda x, y, z: np.testing.assert_allclose( | ||
x, y, err_msg=z, atol=1e-6, rtol=1e-6 | ||
) | ||
if os.getenv("FLAGS_FLAGS_NEW_IR_NO_CHECK", None): | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 这个flag为什么和上面CMake里设置的不一样?而且他能生效? There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 已修改 这里感觉应该单测失败才对 There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 我怀疑这种写法是不是有问题,它会不会总是生效的?可以验证下这个flag对别的单测是不是也起效果了 There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. 确实有问题,if os.getenv("FLAGS_NEW_IR_NO_CHECK", None)只会检查这个环境变量是否存在,如果是0的话也会执行if后面的逻辑,现在改为if os.getenv("FLAGS_NEW_IR_NO_CHECK", None)==“True”只有得到环境变量为“True”才会执行 |
||
check_method = lambda x, y, err_msg: None | ||
|
||
for i in range(len(outs)): | ||
check_method( | ||
|
@@ -3373,6 +3375,9 @@ def _check_ir_grad_output( | |
x, y, err_msg=z, atol=1e-6, rtol=1e-6 | ||
) | ||
|
||
if os.getenv("FLAGS_FLAGS_NEW_IR_NO_CHECK", None): | ||
check_method = lambda x, y, err_msg: None | ||
|
||
for i in range(len(new_gradients)): | ||
check_method( | ||
gradients[i], | ||
|
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.
没有扩展性,建议改为列表实现,类似 #55857
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.
已修改