Skip to content
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.

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

[CI] Allow Limit CPUs in Docker #15668

Merged
merged 1 commit into from
Sep 6, 2023
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
17 changes: 14 additions & 3 deletions docker/bash.sh
Original file line number Diff line number Diff line change
Expand Up @@ -38,9 +38,11 @@ set -euo pipefail
function show_usage() {
cat <<EOF
Usage: docker/bash.sh [-i|--interactive] [--net=host] [-t|--tty]
[--mount MOUNT_DIR] [--repo-mount-point REPO_MOUNT_POINT]
[--dry-run] [--name NAME]
<DOCKER_IMAGE_NAME> [--] [COMMAND]
[--cpus NUM_CPUS] [--mount MOUNT_DIR]
[--repo-mount-point REPO_MOUNT_POINT]
[--dry-run] [--name NAME]
<DOCKER_IMAGE_NAME> [--] [COMMAND]


-h, --help

Expand All @@ -54,6 +56,10 @@ Usage: docker/bash.sh [-i|--interactive] [--net=host] [-t|--tty]

Start the docker session with a pseudo terminal (tty).

--cpus NUM_CPUS

Limit the number of CPU cores to be used.

--net=host

Expose servers run into the container to the host, passing the
Expand Down Expand Up @@ -184,6 +190,11 @@ while (( $# )); do
eval $break_joined_flag
;;

--cpus)
DOCKER_FLAGS+=(--cpus "$2")
shift 2
;;

--net=host)
USE_NET_HOST=true
shift
Expand Down