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.

Sign up for GitHub

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

Already on GitHub? Sign in to your account

Jump to bottom

[NVIDIA] Add FP8 example using TE #3080

Merged
merged 1 commit into from
Mar 10, 2023
Merged

[NVIDIA] Add FP8 example using TE #3080

merged 1 commit into from
Mar 10, 2023

Conversation

ksivaman
Copy link
Contributor

@ksivaman ksivaman commented Mar 9, 2023

馃摑 What does this PR do?

Adds a sample script to use the TransformerEngine API for a simple classification task on the MNIST dataset with FP8 capability.

@ksivaman
Add FP8 example using TE
dd90cc7 
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
@binmakeswell binmakeswell self-requested a review March 10, 2023 02:34
@binmakeswell
Copy link
Member

Hi @ksivaman Thank you very much for your contribution!
We are looking forward to deepening our cooperation with you and NVIDIA.

@binmakeswell binmakeswell changed the title Add FP8 example using TE [NVIDIA] Add FP8 example using TE Mar 10, 2023
@binmakeswell binmakeswell merged commit 65a4dbd into hpcaitech:main Mar 10, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@binmakeswell binmakeswell binmakeswell approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
2 participants
@ksivaman @binmakeswell