Skip to content

Commit f0bb59d

Browse files
牢大牢大
authored andcommitted
test: allow qwen3 tilelet cases on a3
1 parent df70b4e commit f0bb59d

3 files changed

Lines changed: 4 additions & 8 deletions

File tree

.github/workflows/ci.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,7 @@ jobs:
293293
# suite (RUN_ONLY_CASES is empty), skip the non-matching variant based
294294
# on SOC_VERSION to keep the remote validation portable.
295295
A3_ONLY_CASES="partition5d,partition5d_dynamic,mrgsort,tmatmulk_autosync"
296-
A5_ONLY_CASES="partition5d_a5,partition5d_dynamic_a5,mrgsort_a5,tmatmulk_autosync_a5,qwen3_decode_layer_incore_1,qwen3_decode_layer_incore_2,qwen3_decode_layer_incore_10,qwen3_decode_layer_incore_13,qwen3_decode_layer_incore_14"
296+
A5_ONLY_CASES="partition5d_a5,partition5d_dynamic_a5,mrgsort_a5,tmatmulk_autosync_a5"
297297
298298
sv_lc="$(printf '%s' "${SOC_VERSION}" | tr '[:upper:]' '[:lower:]')"
299299
is_a5=0

test/samples/Qwen3Tilelet/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ Qwen3 tilelet PTO kernels generated from `pypto-lib/examples/models/qwen3/qwen3_
22

33
Scope:
44
- compile-regression inputs for `ptoas`
5-
- A5-only kernels; `runop.sh` injects `--pto-arch a5 --pto-level=level3` for this directory unless the caller already overrides `PTOAS_FLAGS`
5+
- tilelet kernels that default to `--pto-arch a5 --pto-level=level3` in `runop.sh`, but can also be compiled on A3 when the caller overrides `PTOAS_FLAGS`
66

77
Notes:
88
- The source PyPTO program lowers to 20 `qwen3_decode_layer_incore_*.pto` fragments; this directory vendors the full emitted `.pto` set regenerated from the tilelet source with `BATCH_TILE=16`.

test/samples/runop.sh

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -154,8 +154,8 @@ process_one_dir() {
154154
use_ptobc_roundtrip=1
155155
fi
156156
# Qwen3 tilelet kernels currently serve as direct ptoas compile-regression
157-
# coverage. They require A5/level3 lowering, but are not expected to
158-
# roundtrip through ptobc yet.
157+
# coverage. Default them to A5/level3 lowering when the caller does not
158+
# provide an explicit arch, but let A3/A5 callers override PTOAS_FLAGS.
159159
if [[ "$A" == "Qwen3Tilelet" ]]; then
160160
use_ptobc_roundtrip=0
161161
fi
@@ -942,10 +942,6 @@ PY
942942
if [[ "$A" == "Qwen3Tilelet" ]]; then
943943
cpp="${out_subdir}/${base}-pto.cpp"
944944
fi
945-
if [[ "$A" == "Qwen3Tilelet" && "$(printf '%s' "$target_arch" | tr '[:upper:]' '[:lower:]')" != "a5" ]]; then
946-
echo -e "${A}(${base}.pto)\tSKIP\trequires --pto-arch=a5"
947-
continue
948-
fi
949945
local sample_use_ptobc_roundtrip="$use_ptobc_roundtrip"
950946

951947
# TODO(ptobc): decode of this regression currently fails with

0 commit comments

Comments
 (0)