[CI][NPU] Fix kimi_k2_6 16p in64k perf test and dsv4-flash testcases (#37760)

2026-09-03 21:57:14 +08:00
parent 9cb38a3d57
commit 3239baef25
6 changed files with 44 additions and 62 deletions
@@ -110,6 +110,9 @@ fi
 # set environment of cann
 . /usr/local/Ascend/cann/set_env.sh
 . /usr/local/Ascend/nnal/atb/set_env.sh
+# Adapt Deepseek-V4-Flash test cases with additional environment variables.
+source /usr/local/Ascend/ascend-toolkit/latest/opp/vendors/customize/bin/set_env.bash || true
+source /usr/local/Ascend/ascend-toolkit/latest/opp/vendors/custom_transformer/bin/set_env.bash || true

 echo "Running test case ${test_case}"
 tc_name=${test_case##*/}