DeepSpeed
FP8 unittest for H100
#3731
Merged

FP8 unittest for H100 #3731

jeffra merged 8 commits into master from jomayeri/te_unittest
jomayeri
jomayeri first file added
dd4921e2
jomayeri full fp8 test
c2818eac
jomayeri te to transformer_engine for precommit
7f263a63
jomayeri jomayeri requested a review from jeffra jeffra 2 years ago
jomayeri jomayeri requested a review from mrwyattii mrwyattii 2 years ago
jomayeri jomayeri requested a review from tjruwase tjruwase 2 years ago
jeffra add module skip flag
ffe4c17c
jomayeri cleaner dtype check
2ffa2893
jomayeri Merge branch 'master' into jomayeri/te_unittest
54fa8e22
jeffra add skips for missing TE and older gpus
255e60e1
jeffra
jeffra approved these changes on 2023-06-09
jeffra jeffra added merge-queue
jomayeri Merge branch 'master' into jomayeri/te_unittest
9a0dd8ca
jomayeri jomayeri enabled auto-merge (squash) 2 years ago
disabled auto-merge 2 years ago
Manually disabled by user
jeffra jeffra merged 6f4fc30b into master 2 years ago
jeffra jeffra deleted the jomayeri/te_unittest branch 2 years ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone