build_wheels_genai_vllm_linux_x86.yml

name: Build FBGEMM GenAI x86 Linux Wheels for vLLM

on:
  pull_request:
  push:
    branches:
      - nightly
      - main
    tags:
      # Release candidate tag look like: v1.11.0-rc1
      - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+
      - v[0-9]+.[0-9]+.[0-9]+
  workflow_dispatch:

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

permissions:
  id-token: write
  contents: read

jobs:
  generate-matrix:
    if: ${{ github.repository_owner == 'pytorch' }}
    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
    with:
      package-type: wheel
      os: linux
      test-infra-repository: pytorch/test-infra
      ##########################################################################
      # Build against torch-2.6.0+cu124 only
      test-infra-ref: fbgemm-release-v1.1.1
      ##########################################################################
      with-cuda: enable
      with-rocm: disable
      with-cpu: disable

  build:
    if: ${{ github.repository_owner == 'pytorch' }}
    needs: generate-matrix
    name: pytorch/FBGEMM
    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
    with:
      repository: pytorch/FBGEMM
      ref: ""
      pre-script: ../.github/scripts/nova_prescript.bash
      post-script: ../.github/scripts/nova_postscript.bash
      smoke-test-script: ""
      env-var-script: .github/scripts/nova_dir.bash
      package-name: fbgemm_gpu
      build-target: genai
      test-infra-repository: pytorch/test-infra
      ##########################################################################
      # Build against torch-2.6.0+cu124 only
      test-infra-ref: fbgemm-release-v1.1.1
      ##########################################################################
      build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
      trigger-event: ${{ github.event_name }}
      timeout: 300