Skip to content

Add xentropy bf16 support#1790

Open
zyeric wants to merge 1 commit intoNVIDIA:masterfrom
zyeric:master
Open

Add xentropy bf16 support#1790
zyeric wants to merge 1 commit intoNVIDIA:masterfrom
zyeric:master

Conversation

@zyeric
Copy link
Copy Markdown

@zyeric zyeric commented Mar 29, 2024

Although the kernel supports dispatching to fp16 and bf16 at the same time, there are two incorrect checkers that enforce the input tensor can only be fp16 in the interface.

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