Skip to content

Commit b5c7fc4

Browse files
牢大牢大
authored andcommitted
test: preserve qwen tilelet level3 under arch overrides
1 parent 6ad8402 commit b5c7fc4

1 file changed

Lines changed: 13 additions & 3 deletions

File tree

test/samples/runop.sh

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -180,6 +180,7 @@ process_one_dir() {
180180

181181
local target_arch="a3"
182182
local has_pto_arch_override=0
183+
local has_pto_level_override=0
183184
if ((${#ptoas_flags[@]})); then
184185
for ((idx=0; idx<${#ptoas_flags[@]}; ++idx)); do
185186
if [[ "${ptoas_flags[idx]}" == "--pto-arch" && $((idx + 1)) -lt ${#ptoas_flags[@]} ]]; then
@@ -188,12 +189,21 @@ process_one_dir() {
188189
elif [[ "${ptoas_flags[idx]}" == --pto-arch=* ]]; then
189190
target_arch="${ptoas_flags[idx]#--pto-arch=}"
190191
has_pto_arch_override=1
192+
elif [[ "${ptoas_flags[idx]}" == "--pto-level" && $((idx + 1)) -lt ${#ptoas_flags[@]} ]]; then
193+
has_pto_level_override=1
194+
elif [[ "${ptoas_flags[idx]}" == --pto-level=* ]]; then
195+
has_pto_level_override=1
191196
fi
192197
done
193198
fi
194-
if [[ "$A" == "Qwen3Tilelet" && $has_pto_arch_override -eq 0 ]]; then
195-
ptoas_flags+=(--pto-arch a5 --pto-level=level3)
196-
target_arch="a5"
199+
if [[ "$A" == "Qwen3Tilelet" ]]; then
200+
if [[ $has_pto_arch_override -eq 0 ]]; then
201+
ptoas_flags+=(--pto-arch a5)
202+
target_arch="a5"
203+
fi
204+
if [[ $has_pto_level_override -eq 0 ]]; then
205+
ptoas_flags+=(--pto-level=level3)
206+
fi
197207
fi
198208
local expected_vec_barrier="pipe_barrier(PIPE_V)"
199209
local skip_vec_barrier=0

0 commit comments

Comments
 (0)