Skip to content

[tritonbench] fix tritonbench noise issue #132

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
xuzhao9 wants to merge 7 commits into from
Closed

[tritonbench] fix tritonbench noise issue #132

Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
f73b04f
fix tritonbench noise issue
xuzhao9 Jan 13, 2026
bfe6aa1
set diff
xuzhao9 Jan 13, 2026
73dc980
fix test
xuzhao9 Jan 13, 2026
6a5ebb0
fix taskset
xuzhao9 Jan 13, 2026
fc9019d
allow core using taskset
xuzhao9 Jan 14, 2026
0551ab9
get allowed core
xuzhao9 Jan 14, 2026
427c09c
get allowed core
xuzhao9 Jan 14, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions .github/workflows/tritonbench.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -168,11 +168,18 @@ jobs:
run: |
set -eux

allowed_core=$(taskset -pc $$ | sed -n "s/.*: \([0-9][0-9]*\)-.*/\1/p")
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).
# see https://github.com/pytorch/pytorch-integration-testing/issues/130
container_name=$(docker run \
${GPU_FLAG:-} \
-e DEVICE_NAME \
-e DEVICE_TYPE \
-e CONDA_ENV \
--cpuset-cpus "${allowed_core}" \
--ipc=host \
--tty \
--detach \
Expand Down
Loading