Merge pull request #2464 from oneapi-src/test-taskset

[benchmarks] pin benchmarks to specific cores

Author: pbalcer

.github/workflows/benchmarks-reusable.yml

@@ -176,11 +176,25 @@ jobs:
     - name: Build UMF
       run: cmake --build ${{github.workspace}}/umf_build -j $(nproc)
 
+    - name: Compute core range
+      run: |
+        # Compute the core range for the first NUMA node, skipping the first 4 cores.
+        # This is to avoid the first cores that the kernel is likely to schedule more work on.
+        CORES=$(lscpu | awk '
+          /NUMA node0 CPU|On-line CPU/ {line=$0}
+          END {
+            split(line, a, " ")
+            split(a[4], b, ",")
+            sub(/^0/, "4", b[1])
+            print b[1]
+          }')
+        echo "CORES=$CORES" >> $GITHUB_ENV
+
     - name: Run benchmarks
       working-directory: ${{ github.workspace }}/ur-repo/
       id: benchmarks
       run: >
-        numactl -N 0 ${{ github.workspace }}/ur-repo/scripts/benchmarks/main.py
+        taskset -c ${{ env.CORES }} ${{ github.workspace }}/ur-repo/scripts/benchmarks/main.py
         ~/bench_workdir
         --sycl ${{ github.workspace }}/sycl_build
         --ur ${{ github.workspace }}/ur_install

scripts/benchmarks/benches/umf.py

@@ -43,6 +43,8 @@ def benchmarks(self) -> list[Benchmark]:
 
 class ComputeUMFBenchmark(Benchmark):
     def __init__(self, bench, name):
+        super().__init__(bench.directory, bench)
+
         self.bench = bench
         self.bench_name = name
         self.oneapi = get_oneapi()
@@ -55,8 +57,6 @@ def __init__(self, bench, name):
 
         self.col_statistics_time = None
 
-        super().__init__(bench.directory)
-
     def bin_args(self) -> list[str]:
         return []