Move PGO collection to the separate script

tools/build/build-many.sh

@@ -62,12 +62,8 @@ do
     echo "Setup build dependencies"
     pip install -r ./.requirements/build.txt -r ./.requirements/test.txt
     # Collect PGO
-    echo "Building profiling version"
-    PGO_DATA_DIR="/tmp/pgo-data/$ABI"
-    RUSTFLAGS="-Cprofile-generate=$PGO_DATA_DIR" python3 -m pip install --editable .
-    echo "Collecting PGO data"
-    PYTHONPATH=src/:$PYTHONPATH python3 ./tools/build/collect-pgo.py
-    $(./tools/build/get-rustup-bin.sh)/llvm-profdata merge -o $PGO_DATA_DIR/merged.profdata $PGO_DATA_DIR
+    PGO_DATA_DIR=`mktemp -d`
+    ./tools/build/build-pgo.sh
     # Build wheel
     echo "Building wheel"
     RUSTFLAGS="-Cprofile-use=$PGO_DATA_DIR/merged.profdata" python3 -m build --wheel

tools/build/build-pgo.sh

@@ -0,0 +1,21 @@
+#!/bin/sh
+# ------------------------------------------------------------------------
+# Build version which collects PGO data
+# ------------------------------------------------------------------------
+# Copyright (C) 2022-24, Gufo Labs
+# ------------------------------------------------------------------------
+
+PGO_DATA_DIR=$1
+if [ "$PGO_DATA_DIR" = "" ]; then
+    echo "PGO data dir must be set"
+    exit 1
+fi
+
+# Collect PGO
+echo "Building profiling version"
+RUSTFLAGS="-Cprofile-generate=$PGO_DATA_DIR" python3 -m pip install --editable .
+echo "Collecting PGO data"
+PYTHONPATH=src/:$PYTHONPATH python3 ./tools/build/pgo-runner.py
+echo "Merging profdata"
+$(./tools/build/get-rustup-bin.sh)/llvm-profdata merge -o $PGO_DATA_DIR/merged.profdata $PGO_DATA_DIR
+echo "PGO profile is written into $PGO_DATA_DIR/merged.profdata"

tools/docs/update-bench-charts.py

@@ -78,7 +78,7 @@ def build_barchart(
     bench: Benchmark, data: List[Tuple[str, float]], scale: str
 ) -> None:
     """
-    Build bar chart into SVG file.
+    Build bar chart into PNG file.
 
     Args:
         bench: Benchmark description.