name: 'e2e test'

on:
  workflow_call:
    inputs:
      vllm:
        required: true
        type: string
      image:
        required: true
        type: string
      type:
        required: true
        type: string
      contains_310:
        required: true
        type: boolean

jobs:
  select-image:
    runs-on: linux-aarch64-a2b3-0
    outputs:
      image: ${{ steps.select.outputs.image }}
      image_a3: ${{ steps.select.outputs.image_a3 }}
      image_310p: ${{ steps.select.outputs.image_310p }}
    steps:
      - name: Select image based on base branch
        id: select
        env:
          IMAGE_NAMESPACE: swr.cn-southwest-2.myhuaweicloud.com/base_image/ascend-ci/vllm-ascend
        run: |
          BRANCH="${{ github.base_ref }}"
          BRANCH_TAG="${BRANCH//\//-}"
          MAIN_IMAGE="${{ inputs.image }}"
          MAIN_IMAGE_A3="${{ inputs.image }}-a3"
          MAIN_IMAGE_310P="${{ inputs.image }}-310p"
          if [ "$BRANCH_TAG" = "main" ]; then
            echo "Target branch is main, using main images: ${MAIN_IMAGE} / ${MAIN_IMAGE_A3} / ${MAIN_IMAGE_310P}"
            echo "image=${MAIN_IMAGE}" >> $GITHUB_OUTPUT
            echo "image_a3=${MAIN_IMAGE_A3}" >> $GITHUB_OUTPUT
            echo "image_310p=${MAIN_IMAGE_310P}" >> $GITHUB_OUTPUT
            exit 0
          fi

          # while target branch is not main
          BRANCH_IMAGE="${IMAGE_NAMESPACE}:${BRANCH_TAG}"
          BRANCH_IMAGE_A3="${IMAGE_NAMESPACE}:${BRANCH_TAG}-a3"
          BRANCH_IMAGE_310P="${IMAGE_NAMESPACE}:${BRANCH_TAG}-310p"
          # Check if branch-specific A2 image exists via IMAGE_NAMESPACE, fallback to main if not
          if docker manifest inspect "${BRANCH_IMAGE}" > /dev/null 2>&1; then
            echo "Using branch image: ${BRANCH_IMAGE}"
            echo "image=${BRANCH_IMAGE}" >> $GITHUB_OUTPUT
          else
            echo "Branch image not found, falling back to ${MAIN_IMAGE}"
            echo "image=${MAIN_IMAGE}" >> $GITHUB_OUTPUT
          fi
          # Check if branch-specific A3 image exists via IMAGE_NAMESPACE, fallback to main if not
          if docker manifest inspect "${BRANCH_IMAGE_A3}" > /dev/null 2>&1; then
            echo "Using branch A3 image: ${BRANCH_IMAGE_A3}"
            echo "image_a3=${BRANCH_IMAGE_A3}" >> $GITHUB_OUTPUT
          else
            echo "Branch A3 image not found, falling back to ${MAIN_IMAGE_A3}"
            echo "image_a3=${MAIN_IMAGE_A3}" >> $GITHUB_OUTPUT
          fi
          # Check if branch-specific 310P image exists via IMAGE_NAMESPACE, fallback to main if not
          if docker manifest inspect "${BRANCH_IMAGE_310P}" > /dev/null 2>&1; then
            echo "Using branch 310P image: ${BRANCH_IMAGE_310P}"
            echo "image_310p=${BRANCH_IMAGE_310P}" >> $GITHUB_OUTPUT
          else
            echo "Branch 310P image not found, falling back to ${MAIN_IMAGE_310P}"
            echo "image_310p=${MAIN_IMAGE_310P}" >> $GITHUB_OUTPUT
          fi

  e2e-light:
    name: singlecard-light
    if: ${{ inputs.type == 'light' }}
    needs: [select-image]
    runs-on: linux-aarch64-a2b3-1
    strategy:
      fail-fast: false
      matrix:
        part: [0]
    container:
      image: ${{ needs.select-image.outputs.image }}
      env:
        VLLM_LOGGING_LEVEL: ERROR
        VLLM_USE_MODELSCOPE: True
        HF_HUB_OFFLINE: 1
    steps:
      - name: Checkout vllm-project/vllm-ascend repo
        uses: actions/checkout@v6
        with:
          fetch-depth: 0
      - name: Check npu and CANN info
        run: |
          npu-smi info
          cat /usr/local/Ascend/ascend-toolkit/latest/"$(uname -i)"-linux/ascend_toolkit_install.info

      - name: Config mirrors
        run: |
          sed -Ei 's@(ports|archive).ubuntu.com@cache-service.nginx-pypi-cache.svc.cluster.local:8081@g' /etc/apt/sources.list
          pip config set global.index-url http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple
          pip config set global.trusted-host cache-service.nginx-pypi-cache.svc.cluster.local
          apt-get update -y
          apt install git -y
          git config --global --add safe.directory "${GITHUB_WORKSPACE}"

      - name: Install system dependencies
        run: |
          apt-get -y install `cat packages.txt`

      - name: Checkout vllm-project/vllm repo
        uses: actions/checkout@v6
        with:
          repository: vllm-project/vllm
          ref: ${{ inputs.vllm }}
          path: ./vllm-empty
          fetch-depth: 1

      - name: Install vllm-project/vllm
        run: |
          pip uninstall -y vllm
          rm -rf /vllm-workspace/vllm
          cp -r ./vllm-empty /vllm-workspace/vllm
          VLLM_TARGET_DEVICE=empty pip install -v -e /vllm-workspace/vllm/

      - name: Install vllm-project/vllm-ascend
        env:
          PIP_EXTRA_INDEX_URL: https://mirrors.huaweicloud.com/ascend/repos/pypi
        run: |
          DEST="/vllm-workspace/vllm-ascend"
          # rebase commits
          git config user.email "action@github.com"
          git config user.name "GitHub Action"
          git fetch origin ${{ github.base_ref }}
          git rebase origin/${{ github.base_ref }}

          IMAGE_SHA=$(git -C "${DEST}" log -1 --format=%H 2>/dev/null || echo "")
          cp -rT . "${DEST}/"
          if [ -n "$IMAGE_SHA" ] && git cat-file -e "${IMAGE_SHA}" 2>/dev/null; then
            C_CHANGES=$(git diff "${IMAGE_SHA}"..HEAD --name-only -- \
              csrc/ cmake/ CMakeLists.txt setup.py requirements.txt requirements-dev.txt)
            echo "[debug] C_CHANGES=${C_CHANGES:-<empty>}"
          else
            echo "[debug] IMAGE_SHA not found in local history (empty or unreachable), forcing reinstall"
            C_CHANGES="yes"
          fi
          
          pip install -r ${DEST}/requirements-dev.txt
          if [ -n "$C_CHANGES" ]; then
            echo "[debug] C code / build changes detected, reinstalling vllm-ascend..."
            pip install -v -e "${DEST}/"
          else
            echo "[debug] No C code / build changes detected, skipping reinstall."
          fi

      - name: Run vllm-project/vllm-ascend test
        working-directory: /vllm-workspace/vllm-ascend
        env:
          PYTORCH_NPU_ALLOC_CONF: max_split_size_mb:256
          VLLM_WORKER_MULTIPROC_METHOD: spawn
        run: |
          python3 .github/workflows/scripts/run_suite.py --suite e2e-singlecard-light --auto-partition-id ${{ matrix.part }} --auto-partition-size 1

  e2e-full:
    name: singlecard-full
    if: ${{ inputs.type == 'full' }}
    needs: [select-image]
    runs-on: linux-aarch64-a2b3-1
    strategy:
      fail-fast: false
      matrix:
        part: [0, 1]
    container:
      image: ${{ needs.select-image.outputs.image }}
      env:
        VLLM_LOGGING_LEVEL: ERROR
        VLLM_USE_MODELSCOPE: True
        HF_HUB_OFFLINE: 1
        MODELSCOPE_HUB_FILE_LOCK: False
    steps:
      - name: Checkout vllm-project/vllm-ascend repo
        uses: actions/checkout@v6
        with:
          fetch-depth: 0

      - name: Check npu and CANN info
        run: |
          npu-smi info
          cat /usr/local/Ascend/ascend-toolkit/latest/"$(uname -i)"-linux/ascend_toolkit_install.info

      - name: Config mirrors
        run: |
          sed -Ei 's@(ports|archive).ubuntu.com@cache-service.nginx-pypi-cache.svc.cluster.local:8081@g' /etc/apt/sources.list
          pip config set global.index-url http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple
          pip config set global.trusted-host cache-service.nginx-pypi-cache.svc.cluster.local
          apt-get update -y
          apt install git -y
          git config --global --add safe.directory "${GITHUB_WORKSPACE}"

      - name: Install system dependencies
        run: |
          apt-get -y install `cat packages.txt`

      - name: Checkout vllm-project/vllm repo
        uses: actions/checkout@v6
        with:
          repository: vllm-project/vllm
          ref: ${{ inputs.vllm }}
          path: ./vllm-empty
          fetch-depth: 1

      - name: Install vllm-project/vllm
        run: |
          pip uninstall -y vllm
          rm -rf /vllm-workspace/vllm
          cp -r ./vllm-empty /vllm-workspace/vllm
          VLLM_TARGET_DEVICE=empty pip install -v -e /vllm-workspace/vllm/

      - name: Install vllm-project/vllm-ascend
        env:
          PIP_EXTRA_INDEX_URL: https://mirrors.huaweicloud.com/ascend/repos/pypi
        run: |
          DEST="/vllm-workspace/vllm-ascend"
          # rebase commits
          git config user.email "action@github.com"
          git config user.name "GitHub Action"
          git fetch origin ${{ github.base_ref }}
          git rebase origin/${{ github.base_ref }}

          IMAGE_SHA=$(git -C "${DEST}" log -1 --format=%H 2>/dev/null || echo "")
          cp -rT . "${DEST}/"
          if [ -n "$IMAGE_SHA" ] && git cat-file -e "${IMAGE_SHA}" 2>/dev/null; then
            C_CHANGES=$(git diff "${IMAGE_SHA}"..HEAD --name-only -- \
              csrc/ cmake/ CMakeLists.txt setup.py requirements.txt requirements-dev.txt)
            echo "[debug] C_CHANGES=${C_CHANGES:-<empty>}"
          else
            echo "[debug] IMAGE_SHA not found in local history (empty or unreachable), forcing reinstall"
            C_CHANGES="yes"
          fi
          pip install -r ${DEST}/requirements-dev.txt
          if [ -n "$C_CHANGES" ]; then
            echo "[debug] C code / build changes detected, reinstalling vllm-ascend..."
            pip install -v -e "${DEST}/"
          else
            echo "[debug] No C code / build changes detected, skipping reinstall."
          fi

      - name: Run e2e test
        working-directory: /vllm-workspace/vllm-ascend
        env:
          VLLM_WORKER_MULTIPROC_METHOD: spawn
          PYTORCH_NPU_ALLOC_CONF: max_split_size_mb:256
        run: |
          python3 .github/workflows/scripts/run_suite.py --suite e2e-singlecard --auto-partition-id ${{ matrix.part }} --auto-partition-size 2

  e2e-2-cards-light:
    name: multicard-2-light
    if: ${{ inputs.type == 'light' }}
    needs: [select-image]
    runs-on: linux-aarch64-a3-2
    strategy:
      fail-fast: false
      matrix:
        part: [0]
    container:
      image: ${{ needs.select-image.outputs.image_a3 }}
      env:
        VLLM_LOGGING_LEVEL: ERROR
        VLLM_USE_MODELSCOPE: True
        HCCL_BUFFSIZE: 1024
        HF_HUB_OFFLINE: 1
    steps:
      - name: Checkout vllm-project/vllm-ascend repo
        uses: actions/checkout@v6
        with:
          fetch-depth: 0
      - name: Check npu and CANN info
        run: |
          npu-smi info
          cat /usr/local/Ascend/ascend-toolkit/latest/"$(uname -i)"-linux/ascend_toolkit_install.info

      - name: Config mirrors
        run: |
          sed -Ei 's@(ports|archive).ubuntu.com@cache-service.nginx-pypi-cache.svc.cluster.local:8081@g' /etc/apt/sources.list
          pip config set global.index-url http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple
          pip config set global.trusted-host cache-service.nginx-pypi-cache.svc.cluster.local
          apt-get update -y
          apt install git -y
          git config --global --add safe.directory "${GITHUB_WORKSPACE}"

      - name: Install system dependencies
        run: |
          apt-get -y install `cat packages.txt`

      - name: Checkout vllm-project/vllm repo
        uses: actions/checkout@v6
        with:
          repository: vllm-project/vllm
          ref: ${{ inputs.vllm }}
          path: ./vllm-empty
          fetch-depth: 1

      - name: Install vllm-project/vllm
        run: |
          pip uninstall -y vllm
          rm -rf /vllm-workspace/vllm
          cp -r ./vllm-empty /vllm-workspace/vllm
          VLLM_TARGET_DEVICE=empty pip install -v -e /vllm-workspace/vllm/

      - name: Install vllm-project/vllm-ascend
        env:
          PIP_EXTRA_INDEX_URL: https://mirrors.huaweicloud.com/ascend/repos/pypi
        run: |
          DEST="/vllm-workspace/vllm-ascend"
          # rebase commits
          git config user.email "action@github.com"
          git config user.name "GitHub Action"
          git fetch origin ${{ github.base_ref }}
          git rebase origin/${{ github.base_ref }}

          IMAGE_SHA=$(git -C "${DEST}" log -1 --format=%H 2>/dev/null || echo "")
          cp -rT . "${DEST}/"
          if [ -n "$IMAGE_SHA" ] && git cat-file -e "${IMAGE_SHA}" 2>/dev/null; then
            C_CHANGES=$(git diff "${IMAGE_SHA}"..HEAD --name-only -- \
              csrc/ cmake/ CMakeLists.txt setup.py requirements.txt requirements-dev.txt)
            echo "[debug] C_CHANGES=${C_CHANGES:-<empty>}"
          else
            echo "[debug] IMAGE_SHA not found in local history (empty or unreachable), forcing reinstall"
            C_CHANGES="yes"
          fi
          pip install -r ${DEST}/requirements-dev.txt
          if [ -n "$C_CHANGES" ]; then
            echo "[debug] C code / build changes detected, reinstalling vllm-ascend..."
            pip install -v -e "${DEST}/"
          else
            echo "[debug] No C code / build changes detected, skipping reinstall."
          fi

      - name: Run vllm-project/vllm-ascend test (light)
        working-directory: /vllm-workspace/vllm-ascend
        env:
          VLLM_WORKER_MULTIPROC_METHOD: spawn
        run: |
          python3 .github/workflows/scripts/run_suite.py --suite e2e-2card-light --auto-partition-id ${{ matrix.part }} --auto-partition-size 1

  e2e-2-cards-full:
    name: multicard-2-full
    if: ${{ inputs.type == 'full' }}
    needs: [select-image]
    runs-on: linux-aarch64-a3-2
    strategy:
      fail-fast: false
      matrix:
        part: [0]
    container:
      image: ${{ needs.select-image.outputs.image_a3 }}
      env:
        VLLM_LOGGING_LEVEL: ERROR
        VLLM_USE_MODELSCOPE: True
        HCCL_BUFFSIZE: 1024
        HF_HUB_OFFLINE: 1
    steps:
      - name: Checkout vllm-project/vllm-ascend repo
        uses: actions/checkout@v6
        with:
          fetch-depth: 0
      - name: Check npu and CANN info
        run: |
          npu-smi info
          cat /usr/local/Ascend/ascend-toolkit/latest/"$(uname -i)"-linux/ascend_toolkit_install.info

      - name: Config mirrors
        run: |
          sed -Ei 's@(ports|archive).ubuntu.com@cache-service.nginx-pypi-cache.svc.cluster.local:8081@g' /etc/apt/sources.list
          pip config set global.index-url http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple
          pip config set global.trusted-host cache-service.nginx-pypi-cache.svc.cluster.local
          apt-get update -y
          apt install git -y
          git config --global --add safe.directory "${GITHUB_WORKSPACE}"

      - name: Install system dependencies
        run: |
          apt-get -y install `cat packages.txt`

      - name: Checkout vllm-project/vllm repo
        uses: actions/checkout@v6
        with:
          repository: vllm-project/vllm
          ref: ${{ inputs.vllm }}
          path: ./vllm-empty
          fetch-depth: 1

      - name: Install vllm-project/vllm
        run: |
          pip uninstall -y vllm
          rm -rf /vllm-workspace/vllm
          cp -r ./vllm-empty /vllm-workspace/vllm
          VLLM_TARGET_DEVICE=empty pip install -v -e /vllm-workspace/vllm/

      - name: Install vllm-project/vllm-ascend
        env:
          PIP_EXTRA_INDEX_URL: https://mirrors.huaweicloud.com/ascend/repos/pypi
        run: |
          DEST="/vllm-workspace/vllm-ascend"
          # rebase commits
          git config user.email "action@github.com"
          git config user.name "GitHub Action"
          git fetch origin ${{ github.base_ref }}
          git rebase origin/${{ github.base_ref }}

          IMAGE_SHA=$(git -C "${DEST}" log -1 --format=%H 2>/dev/null || echo "")
          cp -rT . "${DEST}/"
          if [ -n "$IMAGE_SHA" ] && git cat-file -e "${IMAGE_SHA}" 2>/dev/null; then
            C_CHANGES=$(git diff "${IMAGE_SHA}"..HEAD --name-only -- \
              csrc/ cmake/ CMakeLists.txt setup.py requirements.txt requirements-dev.txt)
            echo "[debug] C_CHANGES=${C_CHANGES:-<empty>}"
          else
            echo "[debug] IMAGE_SHA not found in local history (empty or unreachable), forcing reinstall"
            C_CHANGES="yes"
          fi
          pip install -r ${DEST}/requirements-dev.txt
          if [ -n "$C_CHANGES" ]; then
            echo "[debug] C code / build changes detected, reinstalling vllm-ascend..."
            pip install -v -e "${DEST}/"
          else
            echo "[debug] No C code / build changes detected, skipping reinstall."
          fi

      - name: Run vllm-project/vllm-ascend test (full)
        working-directory: /vllm-workspace/vllm-ascend
        env:
          VLLM_WORKER_MULTIPROC_METHOD: spawn
        run: |
          python3 .github/workflows/scripts/run_suite.py --suite e2e-multicard-2-cards --auto-partition-id ${{ matrix.part }} --auto-partition-size 1

      - name: Run vllm-project/vllm-ascend test (non triton)
        if: ${{ inputs.type == 'full' && matrix.part == 0 }}
        working-directory: /vllm-workspace/vllm-ascend
        env:
          VLLM_WORKER_MULTIPROC_METHOD: spawn
        run: |
          python3 -m pip uninstall -y triton-ascend
          pytest -sv --durations=0 tests/e2e/multicard/2-cards/test_aclgraph_capture_replay.py

  e2e-4-cards-full:
    name: multicard-4-full
    if: ${{ inputs.type == 'full' }}
    needs: [select-image]
    runs-on: linux-aarch64-a3-4
    strategy:
      fail-fast: false
      matrix:
        part: [0]
    container:
      image: ${{ needs.select-image.outputs.image_a3 }}
      env:
        VLLM_LOGGING_LEVEL: ERROR
        VLLM_USE_MODELSCOPE: True
        HF_HUB_OFFLINE: 1
    steps:
      - name: Checkout vllm-project/vllm-ascend repo
        uses: actions/checkout@v6
        with:
          fetch-depth: 0
      - name: Check npu and CANN info
        run: |
          npu-smi info
          cat /usr/local/Ascend/ascend-toolkit/latest/"$(uname -i)"-linux/ascend_toolkit_install.info

      - name: Config mirrors
        run: |
          sed -Ei 's@(ports|archive).ubuntu.com@cache-service.nginx-pypi-cache.svc.cluster.local:8081@g' /etc/apt/sources.list
          pip config set global.index-url http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple
          pip config set global.trusted-host cache-service.nginx-pypi-cache.svc.cluster.local
          apt-get update -y
          apt install git -y
          git config --global --add safe.directory "${GITHUB_WORKSPACE}"

      - name: Install system dependencies
        run: |
          apt-get -y install `cat packages.txt`

      - name: Checkout vllm-project/vllm repo
        uses: actions/checkout@v6
        with:
          repository: vllm-project/vllm
          ref: ${{ inputs.vllm }}
          path: ./vllm-empty
          fetch-depth: 1

      - name: Install vllm-project/vllm
        run: |
          pip uninstall -y vllm
          rm -rf /vllm-workspace/vllm
          cp -r ./vllm-empty /vllm-workspace/vllm
          VLLM_TARGET_DEVICE=empty pip install -v -e /vllm-workspace/vllm/

      - name: Install vllm-project/vllm-ascend
        env:
          PIP_EXTRA_INDEX_URL: https://mirrors.huaweicloud.com/ascend/repos/pypi
        run: |
          DEST="/vllm-workspace/vllm-ascend"
          # rebase commits
          git config user.email "action@github.com"
          git config user.name "GitHub Action"
          git fetch origin ${{ github.base_ref }}
          git rebase origin/${{ github.base_ref }}

          IMAGE_SHA=$(git -C "${DEST}" log -1 --format=%H 2>/dev/null || echo "")
          cp -rT . "${DEST}/"
          if [ -n "$IMAGE_SHA" ] && git cat-file -e "${IMAGE_SHA}" 2>/dev/null; then
            C_CHANGES=$(git diff "${IMAGE_SHA}"..HEAD --name-only -- \
              csrc/ cmake/ CMakeLists.txt setup.py requirements.txt requirements-dev.txt)
            echo "[debug] C_CHANGES=${C_CHANGES:-<empty>}"
          else
            echo "[debug] IMAGE_SHA not found in local history (empty or unreachable), forcing reinstall"
            C_CHANGES="yes"
          fi
          pip install -r ${DEST}/requirements-dev.txt
          if [ -n "$C_CHANGES" ]; then
            echo "[debug] C code / build changes detected, reinstalling vllm-ascend..."
            pip install -v -e "${DEST}/"
          else
            echo "[debug] No C code / build changes detected, skipping reinstall."
          fi

      - name: Run vllm-project/vllm-ascend test for V1 Engine
        working-directory: /vllm-workspace/vllm-ascend
        env:
          VLLM_WORKER_MULTIPROC_METHOD: spawn
        run: |
          python3 .github/workflows/scripts/run_suite.py --suite e2e-multicard-4-cards --auto-partition-id ${{ matrix.part }} --auto-partition-size 1

  e2e_310p:
    name: 310p singlecard
    runs-on: linux-aarch64-310p-1
    if: ${{ inputs.contains_310 }}
    needs: [select-image]
    container:
      image: ${{ needs.select-image.outputs.image_310p }}
      env:
        VLLM_LOGGING_LEVEL: ERROR
        VLLM_USE_MODELSCOPE: True
        HF_HUB_OFFLINE: 1
    steps:
      - name: Checkout vllm-project/vllm-ascend repo
        uses: actions/checkout@v6
        with:
          fetch-depth: 0

      - name: Check npu and CANN info
        run: |
          npu-smi info
          cat /usr/local/Ascend/ascend-toolkit/latest/"$(uname -i)"-linux/ascend_toolkit_install.info

      - name: Config mirrors
        run: |
          sed -Ei 's@(ports|archive).ubuntu.com@cache-service.nginx-pypi-cache.svc.cluster.local:8081@g' /etc/apt/sources.list
          pip config set global.index-url http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple
          pip config set global.trusted-host cache-service.nginx-pypi-cache.svc.cluster.local
          apt-get update -y
          apt install git -y
          git config --global --add safe.directory "${GITHUB_WORKSPACE}"

      - name: Install system dependencies
        run: |
          apt-get -y install `cat packages.txt`

      - name: Checkout vllm-project/vllm repo
        uses: actions/checkout@v6
        with:
          repository: vllm-project/vllm
          ref: ${{ inputs.vllm }}
          path: ./vllm-empty
          fetch-depth: 1

      - name: Install vllm-project/vllm
        run: |
          pip uninstall -y vllm
          rm -rf /vllm-workspace/vllm
          cp -r ./vllm-empty /vllm-workspace/vllm
          VLLM_TARGET_DEVICE=empty pip install -v -e /vllm-workspace/vllm/

      - name: Install vllm-project/vllm-ascend
        env:
          PIP_EXTRA_INDEX_URL: https://mirrors.huaweicloud.com/ascend/repos/pypi
        run: |
          DEST="/vllm-workspace/vllm-ascend"
          # rebase commits
          git config user.email "action@github.com"
          git config user.name "GitHub Action"
          git fetch origin ${{ github.base_ref }}
          git rebase origin/${{ github.base_ref }}

          IMAGE_SHA=$(git -C "${DEST}" log -1 --format=%H 2>/dev/null || echo "")
          cp -rT . "${DEST}/"
          if [ -n "$IMAGE_SHA" ] && git cat-file -e "${IMAGE_SHA}" 2>/dev/null; then
            C_CHANGES=$(git diff "${IMAGE_SHA}"..HEAD --name-only -- \
              csrc/ cmake/ CMakeLists.txt setup.py requirements.txt requirements-dev.txt)
            echo "[debug] C_CHANGES=${C_CHANGES:-<empty>}"
          else
            echo "[debug] IMAGE_SHA not found in local history (empty or unreachable), forcing reinstall"
            C_CHANGES="yes"
          fi
          pip install -r ${DEST}/requirements-dev.txt
          if [ -n "$C_CHANGES" ]; then
            echo "[debug] C code / build changes detected, reinstalling vllm-ascend..."
            pip install -v -e "${DEST}/"
          else
            echo "[debug] No C code / build changes detected, skipping reinstall."
          fi

      - name: Run vllm-project/vllm-ascend test
        working-directory: /vllm-workspace/vllm-ascend
        env:
          PYTORCH_NPU_ALLOC_CONF: max_split_size_mb:256
          VLLM_WORKER_MULTIPROC_METHOD: spawn
        run: |
          pytest -sv --durations=0 tests/e2e/310p/singlecard/test_dense_model_singlecard.py

  e2e_310p-4cards:
    name: 310p multicards 4cards
    runs-on: linux-aarch64-310p-4
    if: ${{ inputs.contains_310 }}
    needs: [select-image]
    container:
      image: ${{ needs.select-image.outputs.image_310p }}
      env:
        VLLM_LOGGING_LEVEL: ERROR
        VLLM_USE_MODELSCOPE: True
        HF_HUB_OFFLINE: 1
    steps:
      - name: Checkout vllm-project/vllm-ascend repo
        uses: actions/checkout@v6
        with:
          fetch-depth: 0

      - name: Check npu and CANN info
        run: |
          npu-smi info
          cat /usr/local/Ascend/ascend-toolkit/latest/"$(uname -i)"-linux/ascend_toolkit_install.info

      - name: Config mirrors
        run: |
          sed -Ei 's@(ports|archive).ubuntu.com@cache-service.nginx-pypi-cache.svc.cluster.local:8081@g' /etc/apt/sources.list
          pip config set global.index-url http://cache-service.nginx-pypi-cache.svc.cluster.local/pypi/simple
          pip config set global.trusted-host cache-service.nginx-pypi-cache.svc.cluster.local
          apt-get update -y
          apt install git -y
          git config --global --add safe.directory "${GITHUB_WORKSPACE}"

      - name: Install system dependencies
        run: |
          apt-get -y install `cat packages.txt`

      - name: Checkout vllm-project/vllm repo
        uses: actions/checkout@v6
        with:
          repository: vllm-project/vllm
          ref: ${{ inputs.vllm }}
          path: ./vllm-empty
          fetch-depth: 1

      - name: Install vllm-project/vllm
        run: |
          pip uninstall -y vllm
          rm -rf /vllm-workspace/vllm
          cp -r ./vllm-empty /vllm-workspace/vllm
          VLLM_TARGET_DEVICE=empty pip install -v -e /vllm-workspace/vllm/

      - name: Install vllm-project/vllm-ascend
        env:
          PIP_EXTRA_INDEX_URL: https://mirrors.huaweicloud.com/ascend/repos/pypi
        run: |
          DEST="/vllm-workspace/vllm-ascend"
          # rebase commits
          git config user.email "action@github.com"
          git config user.name "GitHub Action"
          git fetch origin ${{ github.base_ref }}
          git rebase origin/${{ github.base_ref }}

          IMAGE_SHA=$(git -C "${DEST}" log -1 --format=%H 2>/dev/null || echo "")
          cp -rT . "${DEST}/"
          if [ -n "$IMAGE_SHA" ] && git cat-file -e "${IMAGE_SHA}" 2>/dev/null; then
            C_CHANGES=$(git diff "${IMAGE_SHA}"..HEAD --name-only -- \
              csrc/ cmake/ CMakeLists.txt setup.py requirements.txt requirements-dev.txt)
            echo "[debug] C_CHANGES=${C_CHANGES:-<empty>}"
          else
            echo "[debug] IMAGE_SHA not found in local history (empty or unreachable), forcing reinstall"
            C_CHANGES="yes"
          fi
          pip install -r ${DEST}/requirements-dev.txt
          if [ -n "$C_CHANGES" ]; then
            echo "[debug] C code / build changes detected, reinstalling vllm-ascend..."
            pip install -v -e "${DEST}/"
          else
            echo "[debug] No C code / build changes detected, skipping reinstall."
          fi

      - name: Run vllm-project/vllm-ascend test
        working-directory: /vllm-workspace/vllm-ascend
        env:
          PYTORCH_NPU_ALLOC_CONF: max_split_size_mb:256
          VLLM_WORKER_MULTIPROC_METHOD: spawn
        run: |
          pytest -sv --durations=0 \
          tests/e2e/310p/multicard/test_dense_model_multicard.py \
          tests/e2e/310p/multicard/test_moe_model_multicard.py