-
Notifications
You must be signed in to change notification settings - Fork 1.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
[OpenCL]instance_norm support fp32 #8021
[OpenCL]instance_norm support fp32 #8021
Conversation
Thanks for your contribution! |
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
@@ -139,7 +142,12 @@ class InstanceNormImageCompute : public KernelLite<TARGET(kOpenCL), | |||
|
|||
float epsilon = instance_norm_param_->epsilon; | |||
|
|||
// compute global/local work size | |||
auto device_info = CLRuntime::Global()->GetDeviceInfo(); | |||
int max_work_item_size1 = device_info["CL_DEVICE_MAX_WORK_ITEM_SIZES_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.
max_work_item_size1 变量并没有用到?之前版本中 lws1 是要小于 max_work_item_size1 的
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
* instance_norm support fp32 test=develop
No description provided.