Complete Deprecation of Legacy AQT & Transition to Qwix/FP8#4087

Open

sarunsingla11722 wants to merge 14 commits into

mainfrom

deprecate-aqt-phase2

Commits on Jun 5, 2026

Complete legacy AQT deprecation and transition to Qwix/FP8
sarunsingla11722
committed
Merge remote-tracking branch 'origin/main' into deprecate-aqt-phase2
sarunsingla11722
committed
Fix ShardingTypeError in test_explicit_shard_mode by reverting out_sharding to sharding in RoutedMoE
sarunsingla11722
committed

Commits on Jun 6, 2026

Fix MoE, quantization, distillation, and MaxEngine test failures
sarunsingla11722
committed
Add unit tests to cover missing/partial lines in linears, quantizations, and moe
sarunsingla11722
committed
Resolve merge conflict in train_distill_test.py by matching main
sarunsingla11722
committed
Change use_qwix_quantization to True in test_configure_quantization_paths
sarunsingla11722
committed
Update HLO references for Qwix/FP8 transition
sarunsingla11722
committed
Fix pylint import style violations in quantizations_test.py
sarunsingla11722
committed
Restrict Qwix tiling and group-size overrides to quantized runs in moe.py to prevent VMEM OOM
sarunsingla11722
committed
Fix validation, assertion, and divisibility failures in quantizations_test.py
sarunsingla11722
committed
Add @pytest.mark.tpu_only decorator to TPU-specific MoE coverage tests
sarunsingla11722
committed
Update reference HLO from CI artifact
sarunsingla11722
committed
ci: fix GPU unit and integration test failures on multi-GPU runners
sarunsingla11722
committed