7777 MAX_JOBS=32 python3 -m pip install . --no-build-isolation
7878
7979 - name : FlagTree Build on NVidia (triton_v3.6.x branch)
80- if : ${{ steps.check_backend.outputs.should_skip != 'true' && env.TARGET_BRANCH == 'triton_v3.6.x ' }}
80+ if : ${{ steps.check_backend.outputs.should_skip != 'true' && env.TARGET_BRANCH == 'triton_v3.6.x_test_base_branch ' }}
8181 shell : bash
8282 run : |
8383 set -x
@@ -185,10 +185,10 @@ jobs:
185185 fi
186186
187187 # 安装 FlagGems 依赖
188- # cd "$FLAGGEMS_DIR"
189- # pip install -e . --no-build-isolation 2>/dev/null || true
188+ cd "$FLAGGEMS_DIR"
189+ pip install -e . --no-build-isolation 2>/dev/null || true
190190
191- pip install pyyaml==6.0.2
191+ pip install --no-cache-dir pyyaml==6.0.2
192192 # 生成 pytest marks 参数
193193 MARKS=$(python3 "${SCRIPTS_DIR}/parse_operators.py" "$OPERATORS_YAML")
194194 if [ -z "$MARKS" ]; then
@@ -199,8 +199,8 @@ jobs:
199199 echo "Generated pytest marks: ${MARKS:0:200}..."
200200
201201 # 构建 pytest 命令
202- # PYTEST_CMD="python3 -m pytest tests/ -m \"$MARKS\" --ref=cpu --quick -v --timeout=300 "
203- PYTEST_CMD="python3 -m pytest tests/ -m \"$MARKS\" --ref=cpu -v"
202+ PYTEST_CMD="python3 -m pytest tests/ -m \"$MARKS\" --ref=cpu --quick -v"
203+ # PYTEST_CMD="python3 -m pytest tests/ -m \"$MARKS\" --ref=cpu -v"
204204
205205 # 添加 ignore 参数
206206 if [ -f "$IGNORE_LIST" ]; then # 如果有忽略列表文件就读取它
@@ -213,30 +213,42 @@ jobs:
213213 done < "$IGNORE_LIST"
214214 fi
215215
216+
216217 # 设置超时时间 (55分钟 = 3300秒)
217- TIMEOUT=3600
218+ # TIMEOUT=3600
218219
219220 echo "=========================================="
220221 echo "Running FlagGems accuracy tests"
221- echo "Timeout: ${TIMEOUT}s (~60min)"
222+ # echo "Timeout: ${TIMEOUT}s (~60min)"
222223 echo "Quick mode: enabled"
223224 echo "Reference device: cpu"
224225 echo "=========================================="
225226
226- # 运行测试 (带超时)
227+ # # 运行测试 (带超时)
228+ # cd "$FLAGGEMS_DIR"
229+ # # 如果command 执行失败,就执行后面的代码块
230+ # timeout $TIMEOUT bash -c "$PYTEST_CMD" || {
231+ # EXIT_CODE=$?
232+ # if [ $EXIT_CODE -eq 124 ]; then # timeout 命令的特殊退出码124,如果超时则到这里结束
233+ # echo "Warning: Tests timed after ${TIMEOUT}s"
234+ # # 超时不算失败
235+ # exit 0
236+ # else # 如果pytest命令执行失败
237+ # echo "Warning: Some tests failed with exit code $EXIT_CODE"
238+ # # 测试失败也不阻塞 CI,只发出警告
239+ # exit 0
240+ # fi
241+ # }
242+
243+ # 运行测试 (不带超时)
227244 cd "$FLAGGEMS_DIR"
228- # 如果command 执行失败,就执行后面的代码块
229- timeout $TIMEOUT bash -c "$PYTEST_CMD" || {
245+ # 如果 command 执行失败,就执行后面的代码块
246+ bash -c "$PYTEST_CMD" || {
230247 EXIT_CODE=$?
231- if [ $EXIT_CODE -eq 124 ]; then # timeout 命令的特殊退出码124,如果超时则到这里结束
232- echo "Warning: Tests timed after ${TIMEOUT}s"
233- # 超时不算失败
234- exit 0
235- else # 如果pytest命令执行失败
236- echo "Warning: Some tests failed with exit code $EXIT_CODE"
237- # 测试失败也不阻塞 CI,只发出警告
238- exit 0
239- fi
248+ echo "Warning: Some tests failed with exit code $EXIT_CODE"
249+ # 测试失败也不阻塞 CI,只发出警告
250+ exit 0
240251 }
241- # test only
252+
253+
242254 echo "FlagGems accuracy tests completed successfully"
0 commit comments