-
Notifications
You must be signed in to change notification settings - Fork 242
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
Implement SigmoidFocalLoss operation #3143
base: develop
Are you sure you want to change the base?
Conversation
561ccf3
to
53b5359
Compare
It seems a part of this PR included into #3146 |
587d5e0
to
5427607
Compare
I have updated my code following comments in #3146 . Please take another look at my PR. |
e6ac834
to
e53a4e3
Compare
@junliume can you take a look at Windows build state, plz. I added MIOPEN_INTERNALS_EXPORT but it still fail. |
Set test to Smoke instead of Full
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.
Error log:
3143_log.txt
This PR implement torchvision.ops.sigmoid_focal_loss operation. There is no constraint here, MIOpen is faster than ROCm in all cases.
Average improvement over ROCm
Reduced kernels:
Unreduced kernels:
Detail benchmark
Float32
Float16
BFloat16