[NPU] Fix LLaDA2 MoE OOM after the FRACTAL_NZ cast, re-enabling the NZ speedup (#31772)

This commit is contained in:
wenxuewuhd
2026-07-21 16:09:34 +08:00
committed by GitHub
parent 429f6b6d15
commit c0ed009f5b
2 changed files with 12 additions and 11 deletions
@@ -1,4 +1,3 @@
import os
import unittest
from sglang.test.ascend.gsm8k_ascend_mixin import GSM8KAscendMixin
@@ -31,10 +30,6 @@ class TestLLaDA2Mini(GSM8KAscendMixin, CustomTestCase):
"LowConfidence", # TODO: Add dLLM configurations
"--no-dllm-fdfo", # FDFO (PR #27551) halves single-batch speed on NPU; use sync mode
]
env = {
**os.environ,
"SGLANG_NPU_DISABLE_ACL_FORMAT_WEIGHT": "1", # Need to avoid OOM issue
}
accuracy = 0.88
output_throughput = 70