Repository navigation

chore: add moe tests - #4262

Merged

zewenli98 merged 13 commits into

mainfrom

evanli/hlo-moe-tests

Oct 8, 2026

Merged

chore: add moe tests#4262
zewenli98 merged 13 commits into
mainfrom
evanli/hlo-moe-tests

Commits on Oct 6, 2026
- add moe tests
  zewenli98
  committed
  3233335
- fix tests: disable tf32 when using float32 because small diff may cause topk function to choose different experts
  zewenli98
  committed
  ca5f512
- skip TestFlashAttention on Windows
  zewenli98
  committed
  1a4a120
- test NVIDIA_TF32_OVERRIDE=0
  zewenli98
  committed
  97933ca
- fix: some dev versions of torch 2.12.0 have the limitation but the release version doesn't
  zewenli98
  committed
  f5688c5
- revert NVIDIA_TF32_OVERRIDE=0
  zewenli98
  committed
  ebf377c
- remove precision from run_test() in harness.py
  zewenli98
  committed
  b91d396
- fix tests
  zewenli98
  committed
  916149e
- remove 'precision' in tests
  zewenli98
  committed
  157a49e
- test setting NVIDIA_TF32_OVERRIDE=0
  zewenli98
  committed
  e2529ca
- add xfail to a RTX test
  zewenli98
  committed
  08ac4c9
- revert setting NVIDIA_TF32_OVERRIDE=0
  zewenli98
  committed
  af71406
- clean up legacy
  zewenli98
  committed
  ac39539