updated workflow for pgo

omerbenamram · Apr 3, 2024 · 0748608 · 0748608
1 parent f0ae699
commit 0748608
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/.github/workflows/release.yml b/.github/workflows/release.yml
@@ -142,8 +142,15 @@ jobs:
           echo "RELEASE_VERSION=$release_version" >> $GITHUB_ENV
           echo "release version: $RELEASE_VERSION"
 
+      - name: Build PGO Binary (Linux)
+        if: matrix.build == 'linux'
+        run: |
+          ./build_pgo.sh
+        env:
+          HOME: ${{ github.workspace }}
+
       - name: Build release binary
-        if: matrix.build == 'linux' || matrix.build == 'macos' || matrix.build == 'windows'
+        if: matrix.build == 'macos' || matrix.build == 'windows'
         uses: actions-rs/cargo@v1
         with:
           command: build
@@ -185,3 +192,4 @@ jobs:
           asset_path: ${{ env.ASSET }}
           asset_name: ${{ env.ASSET_NAME }}
           asset_content_type: application/octet-stream
+
diff --git a/build_pgo.sh b/build_pgo.sh
@@ -10,7 +10,7 @@ rm -rf /tmp/pgo-data
 
 PATH=$HOME/.rustup/toolchains/$TOOLCHAIN/lib/rustlib/$TARGET/bin:$PATH
 RUSTFLAGS="-Cprofile-generate=/tmp/pgo-data" \
-    cargo build --release --target $TARGET
+    cargo build --release --target $TARGET --features fast-alloc
 
 echo "Running instrumented binary"
 for i in $(find samples -name "*.evtx"); do
@@ -25,4 +25,4 @@ llvm-profdata merge -o /tmp/pgo-data/merged.profdata /tmp/pgo-data
 
 echo "Building binary with profile data"
 RUSTFLAGS="-Cprofile-use=/tmp/pgo-data/merged.profdata" \
-    cargo build --release --target $TARGET
+    cargo build --release --target $TARGET --features fast-alloc