[JAX] Fix failure on pattern matching of FP8 GEMM when enabling FSDP.#547
Merged
denera merged 8 commits intoNVIDIA:mainfrom mingxu1067:mingh/fix_failure_of_xla_fp8_with_fsdpJan 12, 2024
+212-72
Commits
Commits on Dec 15, 2023
- committed
- committed
- committed
- committed
- committed
- committed
- committed
- committed
