group: forge
steps:
  - name: forge
    wanda: ci/docker/forge.wanda.yaml

  - name: manylinux
    wanda: ci/docker/manylinux.wanda.yaml

  - name: raycudabase
    label: "wanda: ray.py{{matrix.python}}.cu{{matrix.cuda}}.base"
    tags:
      - python_dependencies
      - docker
    wanda: ci/docker/ray.cuda.base.wanda.yaml
    matrix:
      setup:
        python:
          - "3.9"
          - "3.10"
          - "3.11"
          - "3.12"
        cuda:
          - "11.7.1-cudnn8"
          - "11.8.0-cudnn8"
          - "12.1.1-cudnn8"
          - "12.3.2-cudnn9"
    env:
      PYTHON_VERSION: "{{matrix.python}}"
      CUDA_VERSION: "{{matrix.cuda}}"


  - name: raycpubase
    label: "wanda: ray.py{{matrix}}.cpu.base"
    tags:
      - python_dependencies
      - python
      - docker
      - tune
      - serve
    wanda: ci/docker/ray.cpu.base.wanda.yaml
    matrix:
      - "3.9"
      - "3.10"
      - "3.11"
      - "3.12"
    env:
      PYTHON_VERSION: "{{matrix}}"

  - name: ray-mlcudabase
    label: "wanda: ray-ml.py{{matrix.python}}.cu{{matrix.cuda}}.base"
    tags:
      - python_dependencies
      - docker
    wanda: ci/docker/ray-ml.cuda.base.wanda.yaml
    depends_on: raycudabase
    matrix:
      setup:
        python:
          - "3.9"
          - "3.10"
          - "3.11"
        cuda:
          - "12.1.1-cudnn8"
    env:
      PYTHON_VERSION: "{{matrix.python}}"
      CUDA_VERSION: "{{matrix.cuda}}"

  - name: ray-mlcpubase
    label: "wanda: ray-ml.py{{matrix}}.cpu.base"
    tags:
      - python_dependencies
      - docker
    wanda: ci/docker/ray-ml.cpu.base.wanda.yaml
    depends_on: raycpubase
    matrix:
      - "3.9"
      - "3.10"
      - "3.11"
    env:
      PYTHON_VERSION: "{{matrix}}"