tpp-benchmark.yml

name: TPP-MLIR Benchmarks

on:
  workflow_call:
    inputs:
      RUN_SPR_BENCH:
        description: "Run on Sapphire Rapids"
        default: 1
        required: true
        type: string
  push:
    branches: [ "ghactions" ]
  pull_request:
    branches: [ "ghactions" ]

env:
  NPROCS_LIMIT_LINK: 8

jobs:
  Check_LLVM:
    runs-on: self-hosted
    steps:
      - name: Check LLVM
        run: "BUILD=1 scripts/buildkite/check_llvm.sh"

  TPP-MLIR-SPR-BASE:
    runs-on: self-hosted
    needs: Check_LLVM
    steps:
      - name: Sapphire Rapids Base
        run: "${{ secrets.SRUN }} --partition=spr --time=2:00:00 --constraint=\"notrb\" -- \
              'KIND=Release COMPILER=clang LINKER=lld \
              scripts/buildkite/benchmark.sh -b -p'"
    if: ${{ inputs.RUN_SPR_BENCH }} == "1"

  TPP-MLIR-SPR-OMP:
    runs-on: self-hosted
    needs: Check_LLVM
    steps:
      - name: Sapphire Rapids OpenMP
        run: "${{ secrets.SRUN }} --partition=spr --time=2:00:00 --constraint=\"notrb\" -- \
              'KIND=Release COMPILER=clang LINKER=lld \
              scripts/buildkite/benchmark.sh -o'"
    if: ${{ inputs.RUN_SPR_BENCH }} == "1"