ci.yaml

name: "CI"
on:
  pull_request:
    types:
      - opened
      - reopened
      - synchronize
jobs:
  build:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - run: go build ./...
  build-test:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - run: go test -c -tags=e2e ./test/...
  build-image:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - run: docker build --build-arg=KUBERNETES_MINOR_VERSION=latest --file Dockerfile .
  build-image-neuronx:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - run: docker build --file test/images/neuron/Dockerfile .
  build-image-nvidia:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - run: docker build --file test/images/nvidia/Dockerfile .
  build-image-nvidia-training:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - run: |
        docker build --file test/images/nvidia-training/Dockerfile test/images/nvidia-training \
          --build-arg PYTORCH_BUILD_ENV="MAX_JOBS=8 BUILD_TEST=0 USE_FLASH_ATTENTION=0 USE_MEM_EFF_ATTENTION=0 USE_DISTRIBUTED=0"
  build-image-nvidia-inference:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - run: |
        docker build --file test/images/nvidia-inference/Dockerfile test/images/nvidia-inference \
          --build-arg PYTORCH_BUILD_ENV="MAX_JOBS=8 BUILD_TEST=0 USE_FLASH_ATTENTION=0 USE_MEM_EFF_ATTENTION=0 USE_DISTRIBUTED=0"
  build-image-neuron-training:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - run: docker build --file test/images/neuron-training/Dockerfile test/images/neuron-training
  build-image-neuron-inference:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v3
    - run: docker build --file test/images/neuron-inference/Dockerfile test/images/neuron-inference