Skip to content

support bf16#51

Merged
wejoncy merged 2 commits intomainfrom
bf16
Oct 8, 2024
Merged

support bf16#51
wejoncy merged 2 commits intomainfrom
bf16

Conversation

@wejoncy
Copy link
Copy Markdown
Contributor

@wejoncy wejoncy commented Oct 8, 2024

Gracefully support bf16 for all nvidia-GPUs.
For thoce old GPU like V100/T4, we just do the simulation and the computation actually happens in float32.
This would be slower than half then.

@wejoncy wejoncy linked an issue Oct 8, 2024 that may be closed by this pull request
@wejoncy wejoncy merged commit 2d6d252 into main Oct 8, 2024
@wejoncy wejoncy deleted the bf16 branch October 8, 2024 06:47
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.

RuntimeError: un-supported dtype: bfloat16

1 participant