Skip to content

fix: adapt f16 relu#76

Merged
ganler merged 1 commit intoise-uiuc:mainfrom
ganler:fix-relu
Feb 14, 2023
Merged

fix: adapt f16 relu#76
ganler merged 1 commit intoise-uiuc:mainfrom
ganler:fix-relu

Conversation

@ganler
Copy link
Member

@ganler ganler commented Feb 14, 2023

torch.relu do not natively support fp16 but is needed in gradient fn. Do a simple re-casting to make it work.

*should have let it patched together with the last PR but forgot this part. 🤦‍♂️

@ganler ganler merged commit 30743fe into ise-uiuc:main Feb 14, 2023
@ganler ganler deleted the fix-relu branch February 14, 2023 18:17
@ganler ganler mentioned this pull request Feb 19, 2023
1 task
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant