Run pyperformance with --affinity if CPU_AFFINITY (renamed from

PYPERF_CPU_AFFINITY) is set, as well.

Author: Yhg1s

Diff for: bench_runner/scripts/run_benchmarks.py

@@ -82,6 +82,9 @@ def run_benchmarks(
     if extra_args is None:
         extra_args = []
 
+    if (affinity := os.environ.get('CPU_AFFINITY')):
+        extra_args.append(f'--affinity={affinity}')
+
     args = [
         sys.executable,
         "-m",

Diff for: bench_runner/templates/_benchmark.src.yml

@@ -221,7 +221,7 @@ jobs:
       - name: Tune system
         if: ${{ steps.should_run.outputs.should_run != 'false' }}
         run: |
-          sudo LD_LIBRARY_PATH=$LD_LIBRARY_PATH venv/bin/python -m pyperf system ${{ inputs.perf && 'reset' || 'tune ${PYPERF_CPU_AFFINITY:+--affinity="$PYPERF_CPU_AFFINITY"}' }}
+          sudo LD_LIBRARY_PATH=$LD_LIBRARY_PATH venv/bin/python -m pyperf system ${{ inputs.perf && 'reset' || 'tune ${CPU_AFFINITY:+--affinity="$CPU_AFFINITY"}' }}
       - name: Tune for (Linux) perf
         if: ${{ steps.should_run.outputs.should_run != 'false' && inputs.perf }}
         run: |