Convert gsplat cuda code to hip

pfxuan · Mar 3, 2024 · 09d206e · 09d206e
1 parent 39f8c81
commit 09d206e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/.github/workflows/hip.yml b/.github/workflows/hip.yml
@@ -19,7 +19,7 @@ jobs:
         os: [ubuntu-22.04] # [ubuntu-22.04, ubuntu-20.04, ubuntu-18.04]
         arch: [x64] # [x64, x86]
         torch-version: [2.2.1] # [1.12.0, 1.13.0, 2.0.0, 2.1.0, 2.1.1, 2.1.2, 2.2.0, 2.2.1]
-        cuda-version: [12.1.1] # [12.3.1, 12.1.1, 11.8.0, 11.7.1, 11.6.2, 11.5.2,11.4.4, 11.3.1, 11.2.2, 11.1.1, 11.0.3, cpu]
+        cuda-version: [11.8.0] # [12.3.1, 12.1.1, 11.8.0, 11.7.1, 11.6.2, 11.5.2,11.4.4, 11.3.1, 11.2.2, 11.1.1, 11.0.3, cpu]
         rocm-version: [5.7.3] # [5.4.2, 5.6.1, 5.7.3, 6.0.1, 6.0.2]
         cmake-build-type: [Release] # [Debug, ClangTidy]
     env:
@@ -49,6 +49,7 @@ jobs:
             cmake \
             ninja-build \
             libopencv-dev \
+            clang lldb lld \
             wget
           wget -nv https://github.com/ccache/ccache/releases/download/v4.9.1/ccache-4.9.1-linux-x86_64.tar.xz
           sudo tar xf ccache-4.9.1-linux-x86_64.tar.xz -C /usr/bin --strip-components=1 --no-same-owner ccache-4.9.1-linux-x86_64/ccache
@@ -82,6 +83,7 @@ jobs:
           key: rocm-${{matrix.ROCM-VERSION}}-ubuntu
           path: |
             /opt/rocm*
+            /etc/alternatives/rocm
 
       - name: Install ROCm
         if: ${{ steps.rocm-cache.outputs.cache-hit != 'true' }}
@@ -134,7 +136,8 @@ jobs:
           set -x
           source .github/workflows/cuda/${{ runner.os }}-env.sh ${CUDA_VER_SHORT}
           export PATH=$PATH:/opt/rocm/bin
-#          hipify-clang ./vendor/gsplat/backward.cu ./vendor/gsplat/bindings.cu ./vendor/gsplat/forward.cu --cuda-path=${CUDA_HOME}
+          clang --version
+          hipify-clang ./vendor/gsplat/backward.cu ./vendor/gsplat/bindings.cu ./vendor/gsplat/forward.cu --cuda-path=${CUDA_HOME} --print-stats -- -std=c++15
 
       - name: Clean Compiler Cache
         run: |