Optimize GroupNorm - Githubissues

kakaobrain / trident

A performance library for machine learning applications.

https://www.kakaobrain.com

Apache License 2.0

178 stars 11 forks source link

Optimize GroupNorm #146

Closed kakao-steve-ai closed 1 year ago

kakao-steve-ai commented 1 year ago

🙏 Describe the pull request

GroupNorm에서 boundary check 최적화했습니다.

before:

before

after:

after

✅ Checklist

[x] Code follows the project's coding conventions and style.
[x] Tests have been added or updated to cover the changes.
[x] Documentation has been updated, if necessary.