Skip to content

[CI] Add CPU runner support to ssh-runner workflow - #47757

Merged
ydshieh merged 1 commit into
mainfrom
extend_ssh
Aug 4, 2026
Merged

[CI] Add CPU runner support to ssh-runner workflow#47757
ydshieh merged 1 commit into
mainfrom
extend_ssh

Conversation

@ydshieh

@ydshieh ydshieh commented Aug 4, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Adds cpu as a valid runner_type option, mapping to aws-m8i-l-cache
  • Makes num_gpus optional (defaults to none) since CPU runners don't need it
  • If runner_type=a10 is selected without a valid num_gpus, prints a warning and falls back to single
  • Skips the NVIDIA-SMI step for CPU runners

- Add `cpu` as a valid `runner_type` (maps to `aws-m8i-l-cache`)
- Make `num_gpus` optional (defaults to `none`); warn and fall back to `single` if unset for a10
- Skip NVIDIA-SMI step for CPU runners

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
@ydshieh
ydshieh merged commit d31667f into main Aug 4, 2026
28 checks passed
@ydshieh
ydshieh deleted the extend_ssh branch August 4, 2026 08:06
@HuggingFaceDocBuilderDev

Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants