Add fsdp2 fp8 unit tests TE 2.10#492

Open

sudhu2k wants to merge 16 commits intodevfrom

sudhu/FSDP2_unit_tests_fix_2.10

Commits on Mar 17, 2026

Initial commit
sudhu2k
committed
- Updated test functions to include new parameters for FP8 autocasting and refined test case generation for various configurations.

sugovind
authored and
sudhu2k
committed
Refactor quantizer state checks and optimize tensor initialization in FusedAdam. Added debug print for DTensor in MultiTensorApply.
sudhu2k
committed
Refactor assertion function in FP8 tests to use relative and absolute tolerances for tensor comparisons. Updated test logic to accommodate new tolerance parameters for improved accuracy in floating…
sudhu2k
committed
Update test tolerances for FP8 configurations to account for potential differences in gradient calculations. Clean up unused debug print statements in MultiTensorApply and ensure proper newline at …
sudhu2k
committed
Ensure proper newline at the end of the test_torch_fsdp2_fp8.py file by adding a newline character after the pass statement in the test_dummy function.
sudhu2k
committed

Commits on Mar 18, 2026

Commits on Mar 19, 2026

Fix FusedAdam DTensor state initialization for FSDP2
sudhu2k
committed

Commits on Apr 6, 2026

Merge branch 'dev' into sudhu/FSDP2_unit_tests_fix_2.10
sudhu2k
committed

Commits on Apr 7, 2026

Commits on Apr 10, 2026

Refactor FSDP2 FP8 model training and testing setup
sudhu2k
committed

Commits on Apr 13, 2026

Merge remote-tracking branch 'origin/dev' into sudhu/FSDP2_unit_tests_fix_2.10
sudhu2k
committed

Commits on Apr 14, 2026

Enhance SimpleNet model to support linear-only mode
sudhu2k
committed