Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions .github/configs/cuda.yml
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@ display_name: "CUDA Tests"

# Docker image for this hardware
ci_image: localhost:5000/flagscale:cuda12.8.1-cudnn9.7.1-python3.12-torch2.7.1-time2510131515
ci_train_image: localhost:5000/flagscale-train:dev-cu128-py3.12-20260228210721
ci_inference_image: localhost:5000/flagscale-inference:dev-cu128-py3.12-20260302102033
ci_train_image: localhost:5000/flagscale-train:dev-cu128-py3.12
ci_inference_image: localhost:5000/flagscale-inference:dev-cu128-py3.12

# Runner labels for this hardware
runner_labels:
Expand Down
61 changes: 33 additions & 28 deletions .github/workflows/all_tests_common.yml
Original file line number Diff line number Diff line change
Expand Up @@ -115,33 +115,34 @@ jobs:
env_name: ${{ needs.checkout_and_config.outputs.env_name_train }}
env_path: ${{ needs.checkout_and_config.outputs.env_path }}

unit_tests:
needs:
- checkout_and_config
- cli_validation
if: fromJson(needs.checkout_and_config.outputs.device_types)[0] != null
strategy:
fail-fast: false
matrix:
device: ${{ fromJson(needs.checkout_and_config.outputs.device_types) }}
uses: ./.github/workflows/unit_tests_common.yml
name: unit_tests
with:
platform: ${{ inputs.platform }}
device: ${{ matrix.device }}
image: ${{ needs.checkout_and_config.outputs.ci_train_image }}
runs_on: ${{ needs.checkout_and_config.outputs.runs_on }}
container_volumes: ${{ needs.checkout_and_config.outputs.container_volumes }}
container_options: ${{ needs.checkout_and_config.outputs.container_options }}
source_artifact: flagscale-source-${{ github.sha }}
pkg_mgr: ${{ needs.checkout_and_config.outputs.pkg_mgr }}
env_name: ${{ needs.checkout_and_config.outputs.env_name_train }}
env_path: ${{ needs.checkout_and_config.outputs.env_path }}
# unit_tests:
# needs:
# - checkout_and_config
# - cli_validation
# if: fromJson(needs.checkout_and_config.outputs.device_types)[0] != null
# strategy:
# fail-fast: false
# matrix:
# device: ${{ fromJson(needs.checkout_and_config.outputs.device_types) }}
# uses: ./.github/workflows/unit_tests_common.yml
# name: unit_tests
# with:
# platform: ${{ inputs.platform }}
# device: ${{ matrix.device }}
# image: ${{ needs.checkout_and_config.outputs.ci_train_image }}
# runs_on: ${{ needs.checkout_and_config.outputs.runs_on }}
# container_volumes: ${{ needs.checkout_and_config.outputs.container_volumes }}
# container_options: ${{ needs.checkout_and_config.outputs.container_options }}
# source_artifact: flagscale-source-${{ github.sha }}
# pkg_mgr: ${{ needs.checkout_and_config.outputs.pkg_mgr }}
# env_name: ${{ needs.checkout_and_config.outputs.env_name_train }}
# env_path: ${{ needs.checkout_and_config.outputs.env_path }}

functional_tests_train:
needs:
- checkout_and_config
- unit_tests
- cli_validation
# - unit_tests
if: fromJson(needs.checkout_and_config.outputs.train_test_matrix)[0] != null
uses: ./.github/workflows/functional_tests_train.yml
with:
Expand All @@ -159,7 +160,8 @@ jobs:
functional_tests_hetero_train:
needs:
- checkout_and_config
- unit_tests
- cli_validation
# - unit_tests
if: fromJson(needs.checkout_and_config.outputs.hetero_train_test_matrix)[0] != null
uses: ./.github/workflows/functional_tests_hetero_train.yml
with:
Expand All @@ -177,7 +179,8 @@ jobs:
functional_tests_inference:
needs:
- checkout_and_config
- unit_tests
- cli_validation
# - unit_tests
if: fromJson(needs.checkout_and_config.outputs.inference_test_matrix)[0] != null
uses: ./.github/workflows/functional_tests_inference.yml
with:
Expand All @@ -195,7 +198,8 @@ jobs:
functional_tests_serve:
needs:
- checkout_and_config
- unit_tests
- cli_validation
# - unit_tests
if: fromJson(needs.checkout_and_config.outputs.serve_test_matrix)[0] != null
uses: ./.github/workflows/functional_tests_serve.yml
with:
Expand Down Expand Up @@ -232,7 +236,8 @@ jobs:
functional_tests_benchmark:
needs:
- checkout_and_config
- unit_tests
- cli_validation
# - unit_tests
if: fromJson(needs.checkout_and_config.outputs.benchmark_test_matrix)[0] != null
uses: ./.github/workflows/functional_tests_benchmark.yml
with:
Expand All @@ -254,7 +259,7 @@ jobs:
needs:
- checkout_and_config
- cli_validation
- unit_tests
# - unit_tests
- functional_tests_train
- functional_tests_hetero_train
- functional_tests_benchmark
Expand Down
15 changes: 15 additions & 0 deletions .github/workflows/all_tests_cuda.yml
Original file line number Diff line number Diff line change
@@ -1,10 +1,25 @@
name: cuda_tests

on:
# Called by Build Docker Images - CUDA workflow
workflow_call:

push:
branches: ["main"]
paths-ignore:
- 'docker/cuda/**'
- 'docker/build.sh'
- 'tools/install/**'
- 'requirements/**'
- '.github/workflows/build_image_cuda.yml'
pull_request:
branches: ["main"]
paths-ignore:
- 'docker/cuda/**'
- 'docker/build.sh'
- 'tools/install/**'
- 'requirements/**'
- '.github/workflows/build_image_cuda.yml'

concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}-${{ github.actor }}
Expand Down
Loading
Loading