Skip to content
Merged
Changes from all commits
Commits
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
1 change: 1 addition & 0 deletions .buildkite/scripts/hardware_ci/run-xpu-test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,7 @@ docker run \
bash -c '
set -e
echo $ZE_AFFINITY_MASK
pip install tblib==3.1.0
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

high

For better dependency management and to ensure a consistent build environment, it's recommended to add this dependency to a requirements file instead of installing it directly in the CI script. This helps in centralizing dependency management and leverages Docker's layer caching, potentially speeding up the CI process.

Please consider adding tblib==3.1.0 to the appropriate requirements file (e.g., requirements/xpu.txt as used in docker/Dockerfile.xpu) and removing this line from the script.

Copy link
Copy Markdown
Contributor Author

@faaany faaany Sep 11, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

tblib is a test-only dependency. So adding to "xpu.txt" might not be appropriate. But adding an extra "xpu-test.txt" file is also unnecessary for just one dependency. So for now, I think install it directly in the CI script is the most straightforward way.

python3 examples/offline_inference/basic/generate.py --model facebook/opt-125m --block-size 64 --enforce-eager
python3 examples/offline_inference/basic/generate.py --model facebook/opt-125m --block-size 64 -O3 -O.cudagraph_mode=NONE
python3 examples/offline_inference/basic/generate.py --model facebook/opt-125m --block-size 64 --enforce-eager -tp 2 --distributed-executor-backend ray
Expand Down