Skip to content

Commit c3b2f0c

Browse files
authored
Merge branch 'triton_v3.6.x_test_base_branch' into triton_v3.6.x_test_flagGems_branch
2 parents 3f1de89 + 90bcfb3 commit c3b2f0c

1 file changed

Lines changed: 33 additions & 21 deletions

File tree

.github/workflows/flagcicd-hopper-build-and-test.yml

Lines changed: 33 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ jobs:
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

Comments
 (0)