Try making arch configurable

mtreinish · Apr 18, 2024 · 3ba406a · 3ba406a
1 parent 637b998
commit 3ba406a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/build_pgo.sh b/tools/build_pgo.sh
@@ -17,6 +17,8 @@ else
     source build_pgo/bin/activate
 fi
 
+arch=`uname -m`
+
 # Build with instrumentation
 pip install -U -c constraints.txt setuptools-rust wheel setuptools
 RUSTFLAGS="-Cprofile-generate=/tmp/pgo-data" pip install --prefer-binary -c constraints.txt -r requirements-dev.txt -e .
@@ -29,4 +31,4 @@ python tools/pgo_scripts/test_utility_scale.py
 
 deactivate
 
-${HOME}/.rustup/toolchains/*x86_64*/lib/rustlib/x86_64*/bin/llvm-profdata merge -o $merged_path /tmp/pgo-data
+${HOME}/.rustup/toolchains/*$arch*/lib/rustlib/$arch*/bin/llvm-profdata merge -o $merged_path /tmp/pgo-data