.github/workflows/pkgci_regression_test_amdgpu_rocm.yml

# Copyright 2023 The IREE Authors
#
# Licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

name: PkgCI Regression Test (AMDGPU)
on:
  workflow_call:
    inputs:
      artifact_run_id:
        type: string
        default: ""
  workflow_dispatch:
    inputs:
      artifact_run_id:
        type: string
        default: ""

jobs:
  linux_x86_64:
    name: Linux (x86_64)
    runs-on: nodai-amdgpu-w7900-x86-64
    env:
      PACKAGE_DOWNLOAD_DIR: ${{ github.workspace }}/.packages
      IREERS_ARTIFACT_DIR: ${{ github.workspace }}/artifacts
      VENV_DIR: ${{ github.workspace }}/venv
    steps:
      - name: Checking out IREE repository
        uses: actions/checkout@8f4b7f84864484a7bf31766abe9204da3cbe65b3 # v3.5.0
        with:
          submodules: false
      - uses: actions/setup-python@61a6322f88396a6271a6ee3565807d608ecaddd1  # v4.7.0
        with:
          # Must match the subset of versions built in pkgci_build_packages.
          python-version: '3.11'
      - uses: actions/download-artifact@9bc31d5ccc31df68ecc42ccf4149144866c47d8a  # v3.0.2
        with:
          name: linux_x86_64_release_packages
          path: ${{ env.PACKAGE_DOWNLOAD_DIR }}
      - name: Setup venv
        run: |
          ./build_tools/pkgci/setup_venv.py ${VENV_DIR} \
            --artifact-path=${PACKAGE_DOWNLOAD_DIR} \
            --fetch-gh-workflow=${{ inputs.artifact_run_id }}

      # In-tree tests
      - name: Run experimental/regression_suite tests
        run: |
          source ${VENV_DIR}/bin/activate
          export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/rocm/lib:/opt/rocm/hip/lib
          pytest \
            -rA -s -m "plat_rdna3_rocm and presubmit" \
            experimental/regression_suite

      # Disabled while flaky. Might be replaced with HIP tests instead?

      # # Out of tree tests
      # - name: Checking out external TestSuite repository
      #   uses: actions/checkout@8f4b7f84864484a7bf31766abe9204da3cbe65b3 # v3.5.0
      #   with:
      #     repository: nod-ai/SHARK-TestSuite
      #     ref: 65cdac8e1dfab53892337009a9704162e1cf8bfb
      #     path: SHARK-TestSuite
      #     submodules: false
      # - name: Installing external TestSuite Python requirements
      #   run: |
      #     source ${VENV_DIR}/bin/activate
      #     python -m pip install -r SHARK-TestSuite/iree_tests/requirements.txt
      # - name: Run external tests - ONNX test suite
      #   run: |
      #     source ${VENV_DIR}/bin/activate
      #     export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/rocm/lib:/opt/rocm/hip/lib
      #     pytest SHARK-TestSuite/iree_tests/onnx/ \
      #         -n 4 -rpfE --timeout=30 --retries=2 --durations=20 \
      #         --config-files=experimental/regression_suite/external_test_suite/onnx_gpu_rocm_rdna3.json