|
15 | 15 | RUN_NIGHTLY: yes |
16 | 16 | PIPELINE_USAGE_CUTOFF: 5000 |
17 | 17 | SLACK_API_TOKEN: ${{ secrets.SLACK_CIFEEDBACK_BOT_TOKEN }} |
| 18 | + HF_TOKEN: ${{ secrets.DIFFUSERS_HF_HUB_READ_TOKEN }} |
| 19 | + CONSOLIDATED_REPORT_PATH: consolidated_test_report.md |
| 20 | + SHOW_SLOWEST_TESTS: 20 |
| 21 | + HF_REPORT_DATASET: diffusers-ci/nightly-test-reports |
18 | 22 |
|
19 | 23 | jobs: |
20 | 24 | setup_torch_cuda_pipeline_matrix: |
@@ -475,6 +479,65 @@ jobs: |
475 | 479 | run: | |
476 | 480 | pip install slack_sdk tabulate |
477 | 481 | python utils/log_reports.py >> $GITHUB_STEP_SUMMARY |
| 482 | + |
| 483 | + generate_consolidated_report: |
| 484 | + name: Generate Consolidated Test Report |
| 485 | + needs: [ |
| 486 | + run_nightly_tests_for_torch_pipelines, |
| 487 | + run_nightly_tests_for_other_torch_modules, |
| 488 | + run_big_gpu_torch_tests, |
| 489 | + run_nightly_onnx_tests, |
| 490 | + run_nightly_quantization_tests, |
| 491 | + run_flax_tpu_tests |
| 492 | + ] |
| 493 | + if: always() |
| 494 | + runs-on: |
| 495 | + group: aws-general-8-plus |
| 496 | + container: |
| 497 | + image: diffusers/diffusers-pytorch-cpu |
| 498 | + steps: |
| 499 | + - name: Checkout diffusers |
| 500 | + uses: actions/checkout@v3 |
| 501 | + with: |
| 502 | + fetch-depth: 2 |
| 503 | + |
| 504 | + - name: Create reports directory |
| 505 | + run: mkdir -p combined_reports |
| 506 | + |
| 507 | + - name: Download all test reports |
| 508 | + uses: actions/download-artifact@v4 |
| 509 | + with: |
| 510 | + path: artifacts |
| 511 | + |
| 512 | + - name: Prepare reports |
| 513 | + run: | |
| 514 | + # Move all report files to a single directory for processing |
| 515 | + find artifacts -name "*.txt" -exec cp {} combined_reports/ \; |
| 516 | + |
| 517 | + - name: Install dependencies |
| 518 | + run: | |
| 519 | + pip install -e .[test] |
| 520 | + pip install slack_sdk tabulate huggingface_hub |
| 521 | + |
| 522 | + - name: Generate consolidated report |
| 523 | + run: | |
| 524 | + python utils/consolidated_test_report.py \ |
| 525 | + --reports_dir combined_reports \ |
| 526 | + --output_file $CONSOLIDATED_REPORT_PATH \ |
| 527 | + --slack_channel_name diffusers-ci-nightly \ |
| 528 | + --hf_dataset_repo $HF_REPORT_DATASET \ |
| 529 | + --upload_to_hub \ |
| 530 | + --compare_with_previous |
| 531 | + |
| 532 | + - name: Show consolidated report |
| 533 | + run: | |
| 534 | + cat $CONSOLIDATED_REPORT_PATH >> $GITHUB_STEP_SUMMARY |
| 535 | + |
| 536 | + - name: Upload consolidated report |
| 537 | + uses: actions/upload-artifact@v4 |
| 538 | + with: |
| 539 | + name: consolidated_test_report |
| 540 | + path: ${{ env.CONSOLIDATED_REPORT_PATH }} |
478 | 541 |
|
479 | 542 | # M1 runner currently not well supported |
480 | 543 | # TODO: (Dhruv) add these back when we setup better testing for Apple Silicon |
|
0 commit comments