add optional flag for building legion only

flexflow · Jul 24, 2023 · eb04a78 · eb04a78
1 parent 7da2fdd
commit eb04a78
Show file tree

Hide file tree

Showing 2 changed files with 308 additions and 270 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -64,6 +64,42 @@ jobs:
           activate-environment: flexflow
           environment-file: conda/environment.yml
           auto-activate-base: false
+
+      - name: Build Legion
+        env: 
+          FF_GPU_BACKEND: ${{ matrix.gpu_backend }}
+        run: |
+          export PATH=/opt/conda/bin:$PATH
+          export CUDNN_DIR=/usr/local/cuda
+          export CUDA_DIR=/usr/local/cuda
+
+          export FF_BUILD_LEGION=ON
+
+          cores_available=$(nproc --all)
+          n_build_cores=$(( cores_available -1 ))
+
+          mkdir build
+          cd build
+          ../config/config.linux
+          make -j $n_build_cores
+          ../config/config.linux
+          make install
+      
+      - name: Prepare library files 
+        env: 
+          FF_GPU_BACKEND: ${{ matrix.gpu_backend }}
+        run: |
+          echo "Removing unnecessary files..."
+          rm -f build/export/legion/lib/libflexflow.so
+          
+          # extract LEGION tarball file 
+          export LEGION_TARBALL="legion_${{ matrix.os }}_${{ matrix.gpu_backend }}.tar.gz"
+
+          echo "Creating archive $LEGION_TARBALL"
+          tar -zcvf $LEGION_TARBALL build/export/legion/
+          echo "Checking the size of the Legion tarball..."
+          du -h $LEGION_TARBALL
+
 
       - name: Build FlexFlow
         run: |