fix: update

janhq · Jun 21, 2024 · fa5b79e · fa5b79e
1 parent adabd0a
commit fa5b79e
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/.github/workflows/template-quality-gate-pr.yml b/.github/workflows/template-quality-gate-pr.yml
@@ -261,17 +261,24 @@ jobs:
 
           $env:HIP_PATH=$(Resolve-Path 'C:\Program Files\AMD\ROCm\*\bin\clang.exe' | split-path | split-path)
           $env:CMAKE_PREFIX_PATH="${env:HIP_PATH}"
+
+          echo "HIP_PATH=${hipPath}" >> $GITHUB_ENV
+          echo "CMAKE_PREFIX_PATH=${hipPath}" >> $GITHUB_ENV
+          echo "GPU_TARGETS=gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101 gfx1102" >> $GITHUB_ENV
   
       - name: Install HIP SDK and ROCM - Linux
         if: ${{ matrix.rocm && matrix.os == 'Linux' }}
         run: |
           echo "deb http://archive.ubuntu.com/ubuntu jammy main universe" | sudo tee /etc/apt/sources.list.d/jammy-copies.list
-          sudo apt-get update
-          wget https://repo.radeon.com/amdgpu-install/6.1.2/ubuntu/jammy/amdgpu-install_6.1.60102-1_all.deb
-          sudo apt install ./amdgpu-install_6.1.60102-1_all.deb -y
+          wget https://repo.radeon.com/amdgpu-install/6.1.2/ubuntu/focal/amdgpu-install_6.1.60102-1_all.deb
+          sudo apt install ./amdgpu-install_6.1.60102-1_all.deb
           sudo apt update -y
-          sudo apt install amdgpu-dkms rocm -y
-          sudo apt-get install -y build-essential rocblas-dev hipblas-dev
+          sudo apt install amdgpu-dkms rocm build-essential rocblas-dev hipblas-dev -y
+
+          # Set nvcc architecture
+          echo "GPU_TARGETS=gfx803 gfx900 gfx906 gfx908 gfx90a gfx1010 gfx1030 gfx1100 gfx1101 gfx1102" >> $GITHUB_ENV
+          echo "CC=/opt/rocm/llvm/bin/clang" >> $GITHUB_ENV
+          echo "CXX=/opt/rocm/llvm/bin/clang++" >> $GITHUB_ENV
 
       - name: Install choco on Windows
         if: runner.os == 'Windows'