[moe training] update bench script to compare fp8 dynamic quant scaled_grouped_mm fwd+bwd against bf16#2764

Closed

danielvegamyhre wants to merge 6 commits intodanielvegamyhre/stack/38from

danielvegamyhre/stack/39

Commits on Aug 13, 2025

[moe training] update tests for torchtitan moe refactor
danielvegamyhre
committed
[moe training] use custom ops instead of wrap_triton for fp8 rowwise kernels
danielvegamyhre
committed
[moe training] fix scaling type bug; refactor distributed tests
danielvegamyhre
committed
[moe training] use llama4 shapes for kernel benchmarks
danielvegamyhre
committed
[moe training] remove duplicate benchmark script
danielvegamyhre
committed
[moe training] update bench script to compare fp8 dynamic quant scaled_grouped_mm fwd+bwd against bf16
danielvegamyhre
committed