Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adding fwd/bwd cast methods compatible with FP8. #120

Merged
merged 1 commit into from
Jun 28, 2024

Conversation

balancap
Copy link
Contributor

Allowing cast applying only on forward or backward passes respectively. Making it easier to build explicit FP8 code.

@balancap balancap self-assigned this Jun 28, 2024
@balancap balancap added the enhancement New feature or request label Jun 28, 2024
@balancap balancap force-pushed the adding-cast-fwd-bwd-methods branch from d06a8c4 to b064312 Compare June 28, 2024 15:01
Allowing cast applying only on forward or backward passes respectively.
Making it easier to build explicit FP8 code.
@balancap balancap force-pushed the adding-cast-fwd-bwd-methods branch from b064312 to 587d815 Compare June 28, 2024 15:04
@balancap balancap merged commit 1db8b33 into main Jun 28, 2024
2 checks passed
@balancap balancap deleted the adding-cast-fwd-bwd-methods branch June 28, 2024 15:06
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.

1 participant