apache--tvm/.github/workflows/main.yml

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

# GH actions.
# We use it to cover windows and mac builds
# Jenkins is still the primary CI

name: CI

on:
  push:
    branches:
      - main
  pull_request:
    branches:
      - main
  workflow_dispatch:

permissions:
  contents: read

concurrency:
  group: CI-${{ github.event.pull_request.number || github.sha }}
  cancel-in-progress: true

jobs:
  MacOS:
    if: ${{ github.repository == 'apache/tvm' }}
    runs-on: macOS-latest
    steps:
      - uses: actions/checkout@v6.0.2
        with:
          submodules: 'recursive'
      - name: Set up environment
        uses: ./.github/actions/setup
      - name: Detect environment variables
        uses: ./.github/actions/detect-env-vars
        id: env_vars
      - name: Build TVM wheel
        shell: bash -l {0}
        env:
          CMAKE_BUILD_PARALLEL_LEVEL: ${{ steps.env_vars.outputs.cpu_count }}
        run: |
          pip install scikit-build-core
          export CMAKE_ARGS="-DUSE_LLVM=ON -DBUILD_TESTING=OFF"
          pip wheel --no-deps -w dist . -v
      - name: Install TVM from wheel
        shell: bash -l {0}
        run: |
          export TVM_FFI_VERSION="$(python -c 'from importlib.metadata import version; print(version("apache-tvm-ffi"))')"
          pip install ml_dtypes
          # Keep the source-matched tvm-ffi installed by the setup action.
          pip install --no-deps dist/*.whl
          python - <<'PY'
          import os
          from importlib.metadata import requires, version

          assert version("apache-tvm-ffi") == os.environ["TVM_FFI_VERSION"]
          assert "apache-tvm-ffi>=0.1.13" in requires("apache-tvm")
          PY

      - name: Test
        shell: bash -l {0}
        run: >-
          python -m pytest -v tests/python/all-platform-minimal-test
      - name: Minimal Metal Compile-Only
        shell: bash -l {0}
        run: |
          python -m pytest -v -s 'tests/python/codegen/test_gpu_codegen_allreduce.py::test_allreduce_sum_compile'
          python -m pytest -v -s 'tests/python/codegen/test_target_codegen_metal.py::test_func_with_trailing_pod_params'
      - name: Minimal Metal Compile-and-Run
        shell: bash -l {0}
        run: |
          python -m pytest -v -s 'tests/python/codegen/test_target_codegen_metal.py'
          python -m pytest -v -s 'tests/python/codegen/test_target_codegen_gpu_common.py'
          python -m pytest -v -s 'tests/python/codegen/test_gpu_codegen_allreduce.py::test_allreduce_sum[dims0-metal]'

  Windows:
    if: ${{ github.repository == 'apache/tvm' }}
    runs-on: windows-latest
    steps:
      - uses: actions/checkout@v6.0.2
        with:
          submodules: 'recursive'
      - name: Set up environment
        uses: ./.github/actions/setup
      - name: Detect environment variables
        uses: ./.github/actions/detect-env-vars
        id: env_vars
      - name: Install TVM
        shell: cmd /C call {0}
        env:
          CMAKE_BUILD_PARALLEL_LEVEL: ${{ steps.env_vars.outputs.cpu_count }}
        run: |
          pip install scikit-build-core
          set CMAKE_ARGS=-DUSE_LLVM=ON -DBUILD_TESTING=OFF
          pip install --no-deps . -v
      - name: Install test dependencies
        shell: cmd /C call {0}
        run: |
          pip install psutil cloudpickle ml_dtypes numpy packaging scipy tornado typing_extensions
      - name: Test
        shell: cmd /C call {0}
        run: >-
          python -m pytest -v tests/python/all-platform-minimal-test