# yamllint disable rule:line-length
name: "Setup build environment"
description: "Setup the build environment. An action so that it can be shared between in-tree/out-of-tree jobs"

inputs:
  cache-enabled:
    required: true
    default: true

  cache-suffix:
    description: |
      Additional string that is used to compute the ccache hash.
      Different jobs running the action need distinct values for this key,
      but the content is irrelevant.
    required: false
    default: ''
  torch-version:
    description: |
      Additional string to determine wether to test against a stable
      torch release or against the nightly build
    required: false
    default: 'nightly'

runs:
  using: "composite"

  steps:
    - name: Set up Python
      if: ${{ runner.arch == 'X64' }}
      uses: actions/setup-python@v4
      with:
        python-version: '3.11'

    - name: Install MLIR Python depends
      if: ${{ runner.os != 'Linux' }}
      run: |
        python -m pip install -r $GITHUB_WORKSPACE/externals/llvm-project/mlir/python/requirements.txt
      shell: bash

    - name: Install PyTorch nightly depends
      if: ${{ runner.os != 'Linux' }}
      run: |
        python -m pip install -r pytorch-requirements.txt
        python -m pip install -r build-requirements.txt
      shell: bash

    - name: Install prerequisites (Linux)
      if: ${{ runner.os == 'Linux' }}
      run: sudo apt-get install --yes ccache ninja-build
      shell: bash

    - name: Install prerequisites (macOS)
      if: ${{ runner.os == 'macOS' }}
      run: brew install ccache ninja
      shell: bash

    - name: Install prerequisites (Windows)
      if: ${{ runner.os == 'Windows' }}
      run: |
        pip install ninja
        choco install ccache --yes
      shell: bash

    - name: Configure ccache
      if: ${{ inputs.cache-enabled == 'true' }}
      run: |
        rm -rf ${{ github.workspace }}/.ccache
        mkdir -p ${{ github.workspace }}/.ccache
        ccache --set-config "cache_dir=${{ github.workspace }}/.ccache"
        ccache --set-config "compression=true"
        ccache --set-config "max_size=300M"
        ccache --zero-stats
      shell: bash

    - name: Enable ccache
      if: ${{ inputs.cache-enabled == 'true' }}
      uses: actions/cache@v3
      with:
        path: ${{ github.workspace }}/.ccache
        key: ${{ runner.os }}-${{ inputs.cache-suffix }}-${{ github.sha }}
        restore-keys: |
          ${{ runner.os }}-${{ inputs.cache-suffix }}-
          ${{ runner.os }}-