Currently only tests ["scan", "silu_dynamic"], but we have a lot more
|
test_examples_on_x86_910B4: |
|
image: quay.io/ascend/cann:8.5.1-910b-ubuntu22.04-py3.11 |
|
tags: |
|
- docker_npu # Build and test on-device NPU (x86-64 / 910B4) |
|
parallel: |
|
matrix: |
|
- EXAMPLE: ["scan", "silu_dynamic"] |
|
before_script: |
|
- source /usr/local/Ascend/ascend-toolkit/set_env.sh |
|
- pip3 install pyyaml setuptools pytest packaging |
|
- pip3 install -r requirements.txt |
|
script: |
|
- echo "Testing example ${EXAMPLE}" |
|
- make setup_once |
|
- cd examples/jit_cpp/${EXAMPLE} |
|
- export PTO_LIB_PATH=${ASCEND_TOOLKIT_HOME} |
|
- python3 run_${EXAMPLE}.py |
Currently only tests
["scan", "silu_dynamic"], but we have a lot morepto-kernels/.gitlab-ci.yml
Lines 87 to 103 in adffa42