FP8 bwd training (#13121)

2026-03-25 08:39:04 +08:00
parent a0a64c679f
commit 5ebb0c2e0b
3 changed files with 59 additions and 16 deletions
@@ -55,6 +55,7 @@ total_vram = 0

 # Training Related State
 in_training = False
+training_fp8_bwd = False


 def get_supported_float8_types():