buildAndTestCMake.yml

# Copyright 2023 The StableHLO Authors.
#
# Licensed 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
#
#      https://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.

name: CMake Build

on:
  pull_request:
    branches: [ main ]
  push:
    branches: [ main ]
  schedule:
    # Run once every 6hr building against llvm-project@HEAD
    - cron:  '0 */6 * * *'
  workflow_dispatch:

# Ensure that only a single job or workflow using the same
# concurrency group will run at a time. This would cancel
# any in-progress jobs in the same github workflow and github
# ref (e.g. refs/heads/main or refs/pull/<pr_number>/merge).
concurrency:
  group: ${{ github.workflow }}-${{ github.ref }}
  cancel-in-progress: true

jobs:
  cmake-build:
    # Only run scheduled CI on main repo
    if: (github.repository == 'openxla/stablehlo' || github.event_name != 'schedule')
    name: "cmake-build ${{ github.event_name == 'schedule' && '(llvm-project@HEAD)' || ''}}"
    env:
      LLVM_PROJECT_DIR: "llvm-project"
      LLVM_BUILD_DIR: "llvm-build"
      STABLEHLO_BUILD_DIR: "stablehlo-build"
      STABLEHLO_PYTHON_BUILD_DIR: "stablehlo-python-build"
    strategy:
      fail-fast: false
    runs-on: ${{ github.repository == 'openxla/stablehlo' && 'ubuntu-22.04-64core' ||  'ubuntu-22.04'  }}

    steps:
    - name: Checkout StableHLO
      uses: actions/checkout@v4

    - name: Get LLVM Version
      id: llvm-version
      shell: bash
      run: |
        USE_LLVM_HEAD=${{ github.event_name == 'schedule' }}
        if [[ $USE_LLVM_HEAD = true ]]; then
          echo "version=main" >> $GITHUB_OUTPUT
        else
          echo "version=$(cat ${{ github.workspace }}/build_tools/llvm_version.txt)" >> $GITHUB_OUTPUT
        fi;

    - name: Setup workspace
      uses: ./.github/actions/setup-build
      with:
        llvm-version: ${{ steps.llvm-version.outputs.version }}

    - name: Configure and Build LLVM
      shell: bash
      run: |
        ./build_tools/github_actions/ci_build_cmake_llvm.sh "$LLVM_PROJECT_DIR" "$LLVM_BUILD_DIR"
      env:
          CMAKE_BUILD_TYPE: Release
          MLIR_ENABLE_BINDINGS_PYTHON: ON

    - name: Build and Test StableHLO (with AddressSanitizer)
      shell: bash
      run: |
        ./build_tools/github_actions/ci_build_cmake.sh "$LLVM_BUILD_DIR" "$STABLEHLO_BUILD_DIR"
      env:
          CMAKE_BUILD_TYPE: Release
          STABLEHLO_ENABLE_BINDINGS_PYTHON: OFF
          STABLEHLO_ENABLE_SANITIZER: address
          # TODO: remove this once https://github.com/openxla/stablehlo/pull/2318 is fixed
          ASAN_OPTIONS: allow_user_poisoning=false

    - name: Build and Test StableHLO (with Python bindings)
      shell: bash
      run: |
        pip install tensorflow-cpu
        ./build_tools/github_actions/ci_build_cmake.sh "$LLVM_BUILD_DIR" "$STABLEHLO_BUILD_DIR"
      env:
          CMAKE_BUILD_TYPE: Release
          STABLEHLO_ENABLE_BINDINGS_PYTHON: ON
          STABLEHLO_ENABLE_PYTHON_TF_TESTS: ON