From 94eadb157014a2cd34b08390079f859d549fa933 Mon Sep 17 00:00:00 2001
From: Pengfei Xuan <pfxuan@gmail.com>
Date: Sat, 23 Mar 2024 10:35:43 -0400
Subject: [PATCH] Save the build artifacts (latest versions) across all
 platforms

---
 .github/workflows/hip.yml        |  3 ++-
 .github/workflows/ubuntu-cpu.yml |  1 +
 .github/workflows/ubuntu.yml     | 12 +++++++++++-
 .github/workflows/windows.yml    | 12 +++++++++++-
 4 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/hip.yml b/.github/workflows/hip.yml
index 2ffc4c7..962631d 100644
--- a/.github/workflows/hip.yml
+++ b/.github/workflows/hip.yml
@@ -174,7 +174,8 @@ jobs:
             -DOPENSPLAT_BUILD_SIMPLE_TRAINER=ON
           ninja
 
-      - name: Save Converted Artifacts
+      - name: Save Artifacts
+        if: ${{ matrix.os == 'ubuntu-22.04' && matrix.rocm-version == '5.7.1' && matrix.torch-version == '2.2.1' && matrix.cmake-build-type == 'Release' }}
         uses: actions/upload-artifact@v4
         with:
           path: |
diff --git a/.github/workflows/ubuntu-cpu.yml b/.github/workflows/ubuntu-cpu.yml
index fac2f11..b187199 100644
--- a/.github/workflows/ubuntu-cpu.yml
+++ b/.github/workflows/ubuntu-cpu.yml
@@ -92,6 +92,7 @@ jobs:
           ninja
 
       - name: Save Artifacts
+        if: ${{ matrix.os == 'ubuntu-22.04' && matrix.torch-version == '2.2.1' && matrix.cmake-build-type == 'Release' }}
         uses: actions/upload-artifact@v4
         with:
           path: |
diff --git a/.github/workflows/ubuntu.yml b/.github/workflows/ubuntu.yml
index 1144d32..fcb91b4 100644
--- a/.github/workflows/ubuntu.yml
+++ b/.github/workflows/ubuntu.yml
@@ -122,9 +122,19 @@ jobs:
             -DCMAKE_CXX_COMPILER_LAUNCHER=$(which ccache) \
             -DCMAKE_PREFIX_PATH=${{ github.workspace }}/libtorch \
             -DCMAKE_INSTALL_PREFIX=${{github.workspace}}/install \
-            -DCUDA_TOOLKIT_ROOT_DIR=$CUDA_HOME
+            -DCUDA_TOOLKIT_ROOT_DIR=$CUDA_HOME \
+            -DOPENSPLAT_BUILD_SIMPLE_TRAINER=ON
           ninja
 
+      - name: Save Artifacts
+        if: ${{ matrix.os == 'ubuntu-22.04' && matrix.cuda-version == '12.1.1' && matrix.torch-version == '2.2.1' && matrix.cmake-build-type == 'Release' }}
+        uses: actions/upload-artifact@v4
+        with:
+          path: |
+            build/libgsplat.a
+            build/opensplat
+            build/simple_trainer
+
       - name: Clean Compiler Cache
         run: |
           set -x
diff --git a/.github/workflows/windows.yml b/.github/workflows/windows.yml
index d824563..46baeb6 100644
--- a/.github/workflows/windows.yml
+++ b/.github/workflows/windows.yml
@@ -136,9 +136,19 @@ jobs:
             -DCMAKE_CXX_COMPILER_LAUNCHER:FILEPATH=ccache `
             -DCMAKE_PREFIX_PATH=${{ github.workspace }}\libtorch `
             -DOPENCV_DIR=${{ github.workspace }}\OpenCV\opencv\build `
-            -DTORCH_CUDA_ARCH_LIST="${{ env.TORCH_CUDA_ARCH_LIST }}"
+            -DTORCH_CUDA_ARCH_LIST="${{ env.TORCH_CUDA_ARCH_LIST }}" `
+            -DOPENSPLAT_BUILD_SIMPLE_TRAINER=ON
           cmake --build . --parallel --config ${{ matrix.cmake-build-type }}
 
+      - name: Save Artifacts
+        if: ${{ matrix.os == 'windows-2022' && matrix.torch-version == '2.2.1' && matrix.cmake-build-type == 'Release' }}
+        uses: actions/upload-artifact@v4
+        with:
+          path: |
+            build/libgsplat.so
+            build/opensplat
+            build/simple_trainer
+
       - name: Clean Compiler Cache
         shell: pwsh
         run: |