Build seprate binaries for cpu and gpu

livepeer · Sep 21, 2023 · 733bdfa · 733bdfa
1 parent 1894ef7
commit 733bdfa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -71,6 +71,12 @@ jobs:
           echo "GOOS=${{ matrix.platform.name }}" >> $GITHUB_ENV
           echo "GO_BUILD_DIR=lp-builds/" >> $GITHUB_ENV
           echo "PKG_CONFIG_PATH=/github/home/compiled/lib/pkgconfig" >> $GITHUB_ENV
+      - name: Set GPU build environment
+        if: matrix.hardware.type == 'gpu'
+        run: |
+          echo "CPATH=/usr/local/cuda_${{ matrix.arch }}/include" >> $GITHUB_ENV
+          echo "LIBRARY_PATH=/usr/local/cuda_${{ matrix.arch }}/lib64" >> $GITHUB_ENV
+          echo "CGO_LDFLAGS=-L/usr/local/cuda_${{ matrix.arch }}/lib64" >> $GITHUB_ENV
 
       - name: Install dependencies
         run: |