allow core using taskset

Author: xuzhao9

.github/workflows/tritonbench.yml

@@ -168,7 +168,8 @@ jobs:
         run: |
           set -eux
 
-          taskset -pc $$
+          allowed_cores=$(sed -n "s/.*: \([0-9][0-9]*\)-.*/\1/p" $(taskset -pc $$))
+          echo "docker allowed core: ${allowed_core}"
           # using `--cpuset-cpus <SINGLE_CPU_CORE>` is critical for stabilizing the performance
           # on this runner (using SPR CPU). This might be related to task migration/Speed Select, etc.
           # here, we are just picking a random CPU core (core 10).
@@ -178,7 +179,7 @@ jobs:
             -e DEVICE_NAME \
             -e DEVICE_TYPE \
             -e CONDA_ENV \
-            --cpuset-cpus 145 \
+            --cpuset-cpus "${allowed_core}" \
             --ipc=host \
             --tty \
             --detach \