chore: add moe tests - #4262

Open

zewenli98 wants to merge 12 commits into

mainfrom

evanli/hlo-moe-tests

Commits on Jul 17, 2026

add moe tests
zewenli98
committed
44e49ac
fix tests: disable tf32 when using float32 because small diff may cause topk function to choose different experts
zewenli98
committed
f2e0c30
skip TestFlashAttention on Windows
zewenli98
committed
5ef5e48
test NVIDIA_TF32_OVERRIDE=0
zewenli98
committed
7fb48b2
fix: some dev versions of torch 2.12.0 have the limitation but the release version doesn't
zewenli98
committed
1dbbbd3
revert NVIDIA_TF32_OVERRIDE=0
zewenli98
committed
9effad5
remove precision from run_test() in harness.py
zewenli98
committed
9fab25f
fix tests
zewenli98
committed
dda46d0
remove 'precision' in tests
zewenli98
committed
b92fc67
test setting NVIDIA_TF32_OVERRIDE=0
zewenli98
committed
5ef1447
add xfail to a RTX test
zewenli98
committed
a5d28be
revert setting NVIDIA_TF32_OVERRIDE=0
zewenli98
committed
33a6092