Skip to content

enable autocast for bf16 as well#44

Open
furkan-celik wants to merge 1 commit intopixeli99:mainfrom
furkan-celik:main
Open

enable autocast for bf16 as well#44
furkan-celik wants to merge 1 commit intopixeli99:mainfrom
furkan-celik:main

Conversation

@furkan-celik
Copy link
Copy Markdown

In my experiments with A100, I have seen type mismatch errors when bf16 is selected. Which I believe is a bug and this simple change fixed it for me.

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