Skip to content

Commit 3d1a1df

Browse files
author
likai00
committed
add detect for a5 board
1 parent 434066b commit 3d1a1df

2 files changed

Lines changed: 4 additions & 1 deletion

File tree

test/npu_validation/scripts/generate_testcase.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,6 @@
7575
"abs",
7676
"partmin",
7777
"prelu",
78-
"rope_kv_cache",
7978
"rowexpanddiv",
8079
"rowexpandmul",
8180
"rowexpandsub",

test/npu_validation/scripts/run_remote_npu_validation.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,9 @@ if [[ "${RUN_MODE}" == "sim" ]]; then
170170
elif [[ "$(printf '%s' "${_board_chip}" | tr '[:upper:]' '[:lower:]')" == *910b* ]]; then
171171
export PTOAS_BOARD_IS_A3=1
172172
log "Detected A3 board from npu-smi chip name: ${_board_chip}"
173+
elif [[ "$(printf '%s' "${SOC_VERSION} ${SIM_SOC_VERSION}" | tr '[:upper:]' '[:lower:]')" == *950* \
174+
|| "$(printf '%s' "${SOC_VERSION} ${SIM_SOC_VERSION}" | tr '[:upper:]' '[:lower:]')" == *a5* ]]; then
175+
log "Detected A5 target from SOC_VERSION=${SOC_VERSION} SIM_SOC_VERSION=${SIM_SOC_VERSION}; skipping A3 simulator dir fallback."
173176
else
174177
for _sim_dir in "${ASCEND_HOME_PATH}/aarch64-linux/simulator" \
175178
"${ASCEND_HOME_PATH}/x86_64-linux/simulator" \
@@ -184,6 +187,7 @@ else
184187
done
185188
done
186189
fi
190+
export PTOAS_BOARD_IS_A3=0
187191
log "SIM_SOC_VERSION=${SIM_SOC_VERSION}"
188192
log "PTOAS_BOARD_IS_A3=${PTOAS_BOARD_IS_A3}"
189193

0 commit comments

Comments
 (0)