-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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
Simplify the code of conv. #45966
Simplify the code of conv. #45966
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
8dd1870
to
2bf4558
Compare
2bf4558
to
e6ce387
Compare
CudnnAlgorithmsCacheMap& GetConv(const AlgorithmType& algo_type) { | ||
return cudnn_auto_tune_map_[static_cast<int64_t>(algo_type)]; | ||
} | ||
|
||
CudnnAlgorithmsCacheMap& GetConvForward() { |
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.
这些Get 成员方法,感觉不再需要了,应该可以删除:
CudnnAlgorithmsCacheMap& GetConvForward()
CudnnAlgorithmsCacheMap& GetConvBackwardData()
CudnnAlgorithmsCacheMap& GetConvBackwardFilter()
或者再提交1个PR删除这些代码
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.
这几个函数,我是考虑后面有没有可能需要再用到,所以暂时没删。
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.
LGTM
PR types
Function optimization
PR changes
OPs
Describe
OP Benchmark CI运行情况:conv2d_5 forward fail。运行详情如下,所选择的kernel一样,但是执行时间稍有差异,应该是由于波动引起。