|
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