# Copyright Advanced Micro Devices, Inc.
# SPDX-License-Identifier: MIT

name: Setup (Multi-Arch)

on:
  workflow_call:
    inputs:
      build_variant:
        type: string
        default: "release"
      release_type:
        description: >-
          Release type: "" for CI builds, or "dev", "nightly", "prerelease"
          for release builds. Controls artifact bucket selection, IAM role,
          and package versioning.
        type: string
        default: ""
      prerelease_version:
        description: Prerelease version number such as '2'. This gets appended to the computed version after 'rc', like '7.10.0rc2'
        type: string
        default: ""
    outputs:
      enable_build_jobs:
        description: Whether to enable build jobs.
        value: true
      linux_build_config:
        description: JSON object with Linux build configuration, or empty string if skipped.
        value: ${{ jobs.setup.outputs.linux_build_config }}
      windows_build_config:
        description: JSON object with Windows build configuration, or empty string if skipped.
        value: ${{ jobs.setup.outputs.windows_build_config }}
      test_type:
        description: "The test type to run (quick, standard, comprehensive, full)."
        value: 'quick'
      linux_test_labels:
        description: ROCm projects to run Linux tests on. Optional filter.
        value: ${{ jobs.setup.outputs.linux_test_labels }}
      windows_test_labels:
        description: ROCm projects to run Windows tests on. Optional filter.
        value: ${{ jobs.setup.outputs.windows_test_labels }}
      rocm_package_version:
        description: ROCm package version (primarily for Python packages).
        value: ${{ jobs.setup.outputs.rocm_package_version }}

permissions:
  contents: read

jobs:
  setup:
    runs-on: ubuntu-24.04
    env:
      BASE_REF: '0efc7532e43c16b532fdd2887e00aec8eb69e10f'      
    outputs:
      enable_build_jobs: true
      # Temporary fix on gfx94x due to non availability of MI325 machines
      linux_build_config: >-
        {
          "per_family_info": [
            {"amdgpu_family": "gfx94X-dcgpu", "amdgpu_targets": "gfx942", "test-runs-on": "linux-gfx942-1gpu-core42-ossci-rocm", "sanity_check_only_for_family": false},
            {"amdgpu_family": "gfx110X-all", "amdgpu_targets": "", "test-runs-on": "", "sanity_check_only_for_family": true, "bypass_tests_for_releases": true},
            {"amdgpu_family": "gfx90a", "amdgpu_targets": "", "test-runs-on": "", "sanity_check_only_for_family": true, "bypass_tests_for_releases": true}
          ],
          "dist_amdgpu_families": "gfx94X-dcgpu;gfx110X-all;gfx90a",
          "artifact_group": "multi-arch-release",
          "build_variant_label": "release",
          "build_variant_suffix": "",
          "build_variant_cmake_preset": "",
          "expect_failure": false,
          "build_pytorch": true,
          "prebuilt_stages": "",
          "baseline_run_id": ""
        }
      windows_build_config: >-
        {
          "per_family_info": [
            {"amdgpu_family": "gfx1151", "amdgpu_targets": "", "test-runs-on": "windows-gfx1151-gpu-rocm", "benchmark-runs-on": "windows-gfx1151-gpu-rocm"}
          ],
          "dist_amdgpu_families": "gfx1151",
          "artifact_group": "multi-arch-release",
          "build_variant_label": "release",
          "build_variant_suffix": "",
          "build_variant_cmake_preset": "windows-release",
          "expect_failure": false,
          "build_pytorch": true,
          "prebuilt_stages": "",
          "baseline_run_id": ""
        }      
      test_type: 'quick'        
      linux_test_labels: ${{ steps.configure.outputs.linux_test_labels }}
      windows_test_labels: ${{ steps.configure.outputs.windows_test_labels }}
      rocm_package_version: ${{ steps.rocm_package_version.outputs.rocm_package_version }}
    steps:
      - name: Checking out repository
        uses: actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
        with:
          # We need the parent commit to do a diff
          repository: "ROCm/TheRock"
          ref: 'compiler/amd-staging'
          fetch-depth: 2

      - name: SHA of TheRock
        run: |
             git rev-parse HEAD
             git log -1
             
      - name: Set PR_LABELS variable with labels assigned to pull request
        if: ${{ github.event.pull_request }} # only set PR labels var if this is a pull request
        env:
          GITHUB_TOKEN: ${{ github.token }}
          PR_NUMBER: ${{ github.event.number }}
        run: |
          echo "PR_LABELS=$(gh pr view ${PR_NUMBER} --repo ROCm/llvm-project --json labels)" >> $GITHUB_ENV            

      # The script reads GITHUB_EVENT_PATH directly for PR labels and
      # workflow_dispatch inputs — no env var pass-through needed.
      - name: Configuring CI options
        id: configure
        env:
          #INPUT_LINUX_AMDGPU_FAMILIES: ${{ github.event.inputs.linux_amdgpu_families }}
          INPUT_LINUX_AMDGPU_FAMILIES: 'gfx94X-dcgpu,gfx110X-all,gfx1151'
          LINUX_TEST_LABELS: ${{ github.event.inputs.linux_test_labels }}
          LINUX_USE_PREBUILT_ARTIFACTS: ${{ github.event.inputs.linux_use_prebuilt_artifacts }}
          #INPUT_WINDOWS_AMDGPU_FAMILIES: ${{ github.event.inputs.windows_amdgpu_families }}
          INPUT_WINDOWS_AMDGPU_FAMILIES: "gfx1151"
          WINDOWS_TEST_LABELS: ${{ github.event.inputs.windows_test_labels }}
          WINDOWS_USE_PREBUILT_ARTIFACTS: ${{ github.event.inputs.windows_use_prebuilt_artifacts }}          
          BUILD_VARIANT: ${{ inputs.build_variant }}
          RELEASE_TYPE: ${{ inputs.release_type }}            
        run: python ./build_tools/github_actions/configure_multi_arch_ci.py

      # Note: Other scripts treat "dev releases" and "CI builds" differently.
      # Here we set to a "dev" package version even for CI builds:
      #
      # type of build                         | package version | version example
      # ------------------------------------- | --------------- | ------------------
      # ci ("" empty string for release_type) | dev             | 7.10.0.dev0+f689a8
      # dev                                   | dev             | 7.10.0.dev0+f689a8
      # nightly                               | nightly         | 7.10.0a20251021
      # prerelease                            | prerelease      | 7.10.0rc2
      #
      # Other scripts upload "ci" and "dev release" packages to different
      # buckets, use different ccache namespaces, etc.
      - name: Compute package version
        id: rocm_package_version
        run: |
          python ./build_tools/compute_rocm_package_version.py \
            --release-type=${{ inputs.release_type || 'dev' }} \
            --prerelease-version=${{ inputs.prerelease_version }}