Skip to content

WIP: migrate examples to SceneTestCase and upgrade paged attention #7

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
doraemonmj wants to merge 5 commits into from
Closed

WIP: migrate examples to SceneTestCase and upgrade paged attention #7

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
a7765a1
Refactor: migrate A5 examples and tests to SceneTestCase format
Apr 15, 2026
3f8119c
Refactor: migrate remaining TMR examples to tests/st and upgrade page…
Apr 16, 2026
fb39511
fix: Complete the missing function names that were omitted during the…
Apr 16, 2026
949fcfe
Refactor: merge bgemm into benchmark_bgemm and fix a5 paged_attention
Apr 17, 2026
76d9ed1
Refactor: remove legacy run_example.py and ci.py runners
Apr 17, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 5 additions & 22 deletions .github/workflows/ci.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -207,13 +207,10 @@ jobs:
pip install torch --index-url https://download.pytorch.org/whl/cpu
pip install '.[test]'

- name: Run simulation examples (a2a3sim)
run: python ci.py -p a2a3sim -c d96c8784 -t 600 --clone-protocol https

- name: Run pytest scene tests (a2a3sim)
run: |
set +e
pytest examples tests/st --platform a2a3sim --device 0-15 -v --pto-session-timeout 600
pytest examples tests/st --platform a2a3sim --device 0-15 -v --pto-session-timeout 600 --clone-protocol https
rc=$?
if [ $rc -eq 124 ]; then
echo "pytest timed out; retrying with pinned PTO-ISA commit"
Expand Down Expand Up @@ -266,18 +263,15 @@ jobs:
pip install torch --index-url https://download.pytorch.org/whl/cpu
pip install '.[test]'

- name: Run simulation examples (a5sim)
run: python ci.py -p a5sim -c d96c8784 -t 600 --clone-protocol https

- name: Run pytest scene tests (a5sim)
run: |
set +e
pytest examples tests/st --platform a5sim --device 0-15 -v --pto-session-timeout 600
pytest examples tests/st --platform a5sim --device 0-15 -v --pto-session-timeout 600 --clone-protocol https
rc=$?
if [ $rc -eq 124 ]; then
echo "pytest timed out; retrying with pinned PTO-ISA commit"
pytest examples tests/st --platform a5sim --device 0-15 -v \
--pto-session-timeout 600 --pto-isa-commit d96c8784 --clone-protocol https
--pto-session-timeout 600 --pto-isa-commit 3cf259e8 --clone-protocol https
rc=$?
fi
exit $rc
Expand Down Expand Up @@ -319,17 +313,12 @@ jobs:
pip install --upgrade pip
pip install '.[test]'

- name: Run on-device examples (a2a3)
run: |
source .venv/bin/activate
source ${ASCEND_HOME_PATH}/bin/setenv.bash && python ci.py -p a2a3 -d ${DEVICE_RANGE} -c d96c8784 -t 600 --clone-protocol https

- name: Run pytest scene tests (a2a3)
run: |
set +e
source .venv/bin/activate
source ${ASCEND_HOME_PATH}/bin/setenv.bash
python -m pytest examples tests/st --platform a2a3 --device ${DEVICE_RANGE} -v --pto-session-timeout 600
python -m pytest examples tests/st --platform a2a3 --device ${DEVICE_RANGE} -v --pto-session-timeout 600 --clone-protocol https
rc=$?
if [ $rc -eq 124 ]; then
echo "pytest timed out; retrying with pinned PTO-ISA commit"
Expand Down Expand Up @@ -407,15 +396,9 @@ jobs:
source ${ASCEND_HOME_PATH}/bin/setenv.bash
pip install '.[test]'

- name: Run on-device examples (a5)
run: |
source ${ASCEND_HOME_PATH}/bin/setenv.bash
DEVICE_LIST=$(python -c "s,e='${DEVICE_RANGE}'.split('-'); print(','.join(str(i) for i in range(int(s),int(e)+1)))")
task-submit --timeout 1800 --max-time 1800 --device "$DEVICE_LIST" --run "python ci.py -p a5 -d ${DEVICE_RANGE} -c d96c8784 -t 1200 --clone-protocol https"

- name: Run pytest scene tests (a5)
run: |
source ${ASCEND_HOME_PATH}/bin/setenv.bash
DEVICE_LIST=$(python -c "s,e='${DEVICE_RANGE}'.split('-'); print(','.join(str(i) for i in range(int(s),int(e)+1)))")
PYTEST="python -m pytest examples tests/st --platform a5 --device ${DEVICE_RANGE} -v"
PYTEST="python -m pytest examples tests/st --platform a5 --device ${DEVICE_RANGE} -v --clone-protocol https"
task-submit --timeout 1800 --max-time 1800 --device "$DEVICE_LIST" --run "set +e; $PYTEST --pto-session-timeout 1200; rc=\$?; if [ \$rc -eq 124 ]; then echo 'pytest timed out; retrying with pinned PTO-ISA commit'; $PYTEST --pto-session-timeout 1200 --pto-isa-commit d96c8784 --clone-protocol https; rc=\$?; fi; exit \$rc"
Loading
Loading