Skip to content

Conversation

@vinx13
Copy link
Member

@vinx13 vinx13 commented Apr 9, 2024

This added missing handling of broadcast and const for fp8 codegen. It relies on the cast operator from float because there's no direct ways to create fp8 literal.

cc @csullivan @tqchen

@vinx13 vinx13 changed the title Feat/fp8 broadcast [Codegen, CUDA] Add handling of fp8 broadcast / const Apr 9, 2024
@github-actions github-actions bot requested review from csullivan and tqchen April 9, 2024 22:57
Copy link
Contributor

@csullivan csullivan left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thank you @vinx13. One nit, up to you if you want the documentation or not.

@tqchen tqchen merged commit 6748215 into apache:main Apr 10, 2024
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.

3 participants