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

name: Multi-Arch Release Windows

on:
  workflow_call:
    inputs:
      build_config:
        type: string
        description: JSON object with build configuration for this platform.
      rocm_package_version:
        type: string
      test_type:
        type: string
      release_type:
        description: 'Release type: "dev", "nightly", or "prerelease".'
        type: string
        required: true
      repository:
        description: "Repository to checkout. Defaults to github.repository."
        type: string
      ref:
        description: "Branch, tag, or SHA to checkout. Defaults to the triggering ref."
        type: string
      build_pytorch:
        description: "Build PyTorch wheels (defaults to true for workflow_call)"
        type: boolean
        default: true
      python_version:
        description: "Single Python version for wheel builds (empty for full matrix)"
        type: string
        default: ""

permissions:
  contents: read

jobs:
  build_artifacts:
    name: Build Artifacts
    uses: ./.github/workflows/multi_arch_build_windows.yml
    secrets: inherit
    with:
      matrix_per_family_json: ${{ toJSON(fromJSON(inputs.build_config).per_family_info) }}
      dist_amdgpu_families: ${{ fromJSON(inputs.build_config).dist_amdgpu_families }}
      artifact_group: ${{ fromJSON(inputs.build_config).artifact_group }}
      build_variant_label: ${{ fromJSON(inputs.build_config).build_variant_label }}
      build_variant_cmake_preset: ${{ fromJSON(inputs.build_config).build_variant_cmake_preset }}
      build_variant_suffix: ${{ fromJSON(inputs.build_config).build_variant_suffix }}
      build_runs_on: ${{ fromJSON(inputs.build_config).build_runs_on }}
      rocm_package_version: ${{ inputs.rocm_package_version }}
      release_type: ${{ inputs.release_type }}
      repository: ${{ inputs.repository }}
      ref: ${{ inputs.ref }}
    permissions:
      contents: read
      id-token: write

  build_tarballs:
    needs: [build_artifacts]
    name: Build Tarballs
    uses: ./.github/workflows/multi_arch_build_tarballs.yml
    with:
      dist_amdgpu_families: ${{ fromJSON(inputs.build_config).dist_amdgpu_families }}
      platform: windows
      package_version: ${{ inputs.rocm_package_version }}
      release_type: ${{ inputs.release_type }}
      repository: ${{ inputs.repository }}
      ref: ${{ inputs.ref }}
    permissions:
      contents: read
      id-token: write

  build_python_packages:
    needs: [build_artifacts]
    name: Build Python Packages
    uses: ./.github/workflows/build_windows_python_packages.yml
    with:
      artifact_group: ${{ fromJSON(inputs.build_config).artifact_group }}
      amdgpu_families: ${{ fromJSON(inputs.build_config).dist_amdgpu_families }}
      linux_amdgpu_families: ${{ fromJSON(inputs.build_config).linux_amdgpu_families }}
      windows_amdgpu_families: ${{ fromJSON(inputs.build_config).windows_amdgpu_families }}
      multiarch_index: true
      package_version: ${{ inputs.rocm_package_version }}
      release_type: ${{ inputs.release_type }}
      repository: ${{ inputs.repository }}
      ref: ${{ inputs.ref }}
    permissions:
      contents: read
      id-token: write

  publish_to_release_buckets:
    needs: [build_tarballs, build_python_packages]
    name: Publish to Release Buckets
    runs-on: ubuntu-24.04
    permissions:
      id-token: write
    steps:
      - name: Checkout
        uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
        with:
          repository: ${{ inputs.repository || github.repository }}
          ref: ${{ inputs.ref }}

      - name: Setting up Python
        uses: actions/setup-python@5fda3b95a4ea91299a34e894583c3862153e4b97 # v7.0.0
        with:
          python-version: "3.12"

      - name: Install Python requirements
        run: pip install -r requirements.txt

      - name: Configure AWS Credentials
        uses: ./.github/actions/configure_aws_artifacts_credentials
        with:
          release_type: ${{ inputs.release_type }}

      - name: Copy to release buckets
        run: |
          python build_tools/github_actions/publish_rocm_to_release_buckets.py \
            --run-id="${{ github.run_id }}" \
            --platform=windows \
            --release-type="${{ inputs.release_type }}" \
            --kpack-split="${{ needs.build_python_packages.outputs.kpack_split }}"

  trigger_test_artifacts_per_family:
    needs: [build_artifacts]
    name: Trigger Test Artifacts (${{ matrix.family_info.amdgpu_family }})
    if: ${{ !failure() && !cancelled() }}
    runs-on: ubuntu-24.04
    permissions:
      actions: write
    strategy:
      fail-fast: false
      matrix:
        family_info: ${{ fromJSON(inputs.build_config).per_family_info }}
    steps:
      - name: Trigger test_artifacts
        if: ${{ matrix.family_info.test-runs-on != '' }}
        uses: benc-uk/workflow-dispatch@31e2b3319479a63f0ab15bf800eff9e913504e26 # v1.3.2
        with:
          workflow: test_artifacts.yml
          inputs: |
            { "artifact_group": "${{ matrix.family_info.amdgpu_family }}",
              "artifact_run_id": "${{ github.run_id }}",
              "amdgpu_families": "${{ matrix.family_info.amdgpu_family }}",
              "amdgpu_targets": "${{ matrix.family_info.amdgpu_targets }}",
              "test_runs_on": "${{ matrix.family_info.test-runs-on }}",
              "test_type": "${{ inputs.test_type }}",
              "release_type": "${{ inputs.release_type }}",
              "repository": "${{ inputs.repository }}",
              "ref": "${{ inputs.ref }}"
            }

  # TODO: Consider switch pulling ROCm packages from the CDN instead of the artifact
  #   bucket if the job order is not altered.
  trigger_release_pytorch_wheels:
    needs: [build_artifacts, build_python_packages]
    name: Trigger Release PyTorch Wheels
    if: ${{ inputs.build_pytorch != false && fromJSON(inputs.build_config).build_pytorch == true }}
    runs-on: ubuntu-24.04
    permissions:
      actions: write
    steps:
      - name: Trigger PyTorch Build
        uses: benc-uk/workflow-dispatch@31e2b3319479a63f0ab15bf800eff9e913504e26 # v1.3.2
        with:
          workflow: multi_arch_release_windows_pytorch_wheels.yml
          inputs: |
            { "amdgpu_families": "${{ fromJSON(inputs.build_config).dist_amdgpu_families }}",
              "rocm_version": "${{ inputs.rocm_package_version }}",
              "rocm_package_find_links_url": "${{ needs.build_python_packages.outputs.package_find_links_url }}",
              "release_type": "${{ inputs.release_type }}",
              "ref": "${{ inputs.ref || '' }}",
              "python_version": "${{ inputs.python_version }}"
            }