Skip to content

feat: Add BF16 burn if GPU supports it.#5

Merged
Hugoch merged 3 commits intomainfrom
feat/use-bf16-when-available
Mar 4, 2025
Merged

feat: Add BF16 burn if GPU supports it.#5
Hugoch merged 3 commits intomainfrom
feat/use-bf16-when-available

Conversation

@Hugoch
Copy link
Member

@Hugoch Hugoch commented Mar 4, 2025

  • Add support for BF16 to use Tensor cores on GPUs that support it (>8.0).

Fixes #4 and #2

@Hugoch Hugoch added the enhancement New feature or request label Mar 4, 2025
@Hugoch Hugoch self-assigned this Mar 4, 2025
@Hugoch Hugoch merged commit c83e0ef into main Mar 4, 2025
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Use tensor cores to ensure max power consumption

1 participant