Commit d76b181

committed

Make MoE models non-strict tracing friendly

1 parent 0187d5f commit d76b181Copy full SHA for d76b181

2 files changed

+471

-3

lines changed

torchtitan
- experiments/graph_trainer/tests
  - test_trace_module.py
- models/common/moe
  - kernels.py

+471

-3

lines changed

Comments

(0)