Skip to content

[Model] Add aquila 3b configuration file #198

[Model] Add aquila 3b configuration file

[Model] Add aquila 3b configuration file #198

Workflow file for this run

name: All Tests
on:
push:
branches: [ "main" ]
pull_request:
branches: [ "main" ]
concurrency:
group: ${{ github.event.pull_request.number }}-${{ github.actor }}
cancel-in-progress: true
jobs:
# Megatron Unit Tests with Matrix
megatron-unit-tests:
uses: ./.github/workflows/unit-tests.yml
strategy:
matrix:
subset:
- data
- dist_checkpointing
- distributed
- fusions
- inference
- models
- pipeline_parallel
- tensor_parallel
- transformer/moe
- transformer
- ./
name: "megatron-${{ matrix.subset == './' && 'root' || matrix.subset }}"
with:
backend: megatron
subset: ${{ matrix.subset }}
# Flagscale Unit Tests with Matrix
flagscale-unit-tests:
uses: ./.github/workflows/unit-tests.yml
strategy:
matrix:
subset:
- runner
- ./
name: "flagscale-${{ matrix.subset == './' && 'root' || matrix.subset }}"
with:
backend: flagscale
subset: ${{ matrix.subset }}
# Functional Tests with Model and Type Matrix
functional-tests-train:
needs:
- megatron-unit-tests
- flagscale-unit-tests
uses: ./.github/workflows/functional-tests.yml
strategy:
matrix:
model:
- aquila
- mixtral
name: "train-${{ matrix.model }}"
with:
model: ${{ matrix.model }}
type: train
functional-tests-hetero:
needs: functional-tests-train
uses: ./.github/workflows/functional-tests.yml
strategy:
matrix:
model:
- aquila
name: "hetero_train-${{ matrix.model }}"
with:
model: ${{ matrix.model }}
type: hetero_train
# Megatron Coverage Test
megatron-coverage-test:
needs: functional-tests-hetero
if: ${{ github.event_name == 'pull_request' }}
uses: ./.github/workflows/coverage-tests.yml
with:
backend: megatron
# Flagscale Coverage Test
flagscale-coverage-test:
needs: functional-tests-hetero
if: ${{ github.event_name == 'pull_request' }}
uses: ./.github/workflows/coverage-tests.yml
with:
backend: flagscale
# Check All Tests
all-tests:
needs:
- megatron-unit-tests
- flagscale-unit-tests
- functional-tests-train
- functional-tests-hetero
- megatron-coverage-test
- flagscale-coverage-test
runs-on: ubuntu-latest
steps:
- name: All Tests Completed
run: echo "All tests completed successfully!"
# Add in the feature for inference
# functional-tests-inference:
# needs: functional-tests-hetero
# uses: ./.github/workflows/functional-tests.yml
# strategy:
# matrix:
# model:
# - from_vllm
# name: "inference-${{ matrix.model }}"
# with:
# model: ${{ matrix.model }}
# type: inference