|
60 | 60 | uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
|
61 | 61 | strategy:
|
62 | 62 | matrix:
|
63 |
| - model: [add] |
| 63 | + model: [add, softmax, mv2] |
64 | 64 | fail-fast: false
|
65 | 65 | with:
|
66 | 66 | runner: linux.2xlarge
|
|
72 | 72 | MODEL_NAME=${{ matrix.model }}
|
73 | 73 | CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
|
74 | 74 | conda activate "${CONDA_ENV}"
|
| 75 | + if [[ ${{ matrix.model}} == "add" ]]; then |
| 76 | + SIM_LIMIT_SEC=60 |
| 77 | + elif [[ ${{ matrix.model}} == "softmax" ]]; then |
| 78 | + SIM_LIMIT_SEC=60 |
| 79 | + elif [[ ${{ matrix.model}} == "mv2" ]]; then |
| 80 | + SIM_LIMIT_SEC=5000 |
| 81 | + else |
| 82 | + echo "Failed unsupported model selection ${{ matrix.model }}" |
| 83 | + exit 1 |
| 84 | + fi |
75 | 85 |
|
76 | 86 | source .ci/scripts/utils.sh
|
77 | 87 | source .ci/scripts/zephyr-utils.sh
|
@@ -118,24 +128,22 @@ jobs:
|
118 | 128 | -C mps3_board.uart0.out_file='sim.out' \
|
119 | 129 | -C cpu0.CFGITCMSZ=15 \
|
120 | 130 | -C cpu0.CFGDTCMSZ=15 \
|
121 |
| - --simlimit 120 |
| 131 | + --simlimit ${SIM_LIMIT_SEC} |
122 | 132 |
|
123 | 133 | # Disable exit on error
|
124 | 134 | set +e
|
125 | 135 | # Report failure if any of the ouptut verification checks fail
|
126 |
| - # store 0 if found (failure), 1 if not (success) |
127 | 136 | grep -qF "ERROR" sim.out
|
128 |
| - exit_status=$? |
| 137 | + exit_status=$? #store 0 if found (failure), 1 if not (success) |
129 | 138 | if [[ "$exit_status" -eq "0" ]]; then
|
130 | 139 | cat sim.out
|
131 | 140 | set -e
|
132 | 141 | exit 1
|
133 | 142 | fi
|
134 | 143 |
|
135 | 144 | # Report fail if simulation does not complete successfully
|
136 |
| - # store 0 if found (success), 1 if not (failure) |
137 | 145 | grep -qF "SUCCESS: Program complete, exiting." sim.out
|
138 |
| - exit_status=$? |
| 146 | + exit_status=$? #store 0 if found (success), 1 if not (failure) |
139 | 147 | if [[ "$exit_status" -eq "1" ]]; then
|
140 | 148 | cat sim.out
|
141 | 149 | set -e
|
|
0 commit comments