Skip to content

switch docs builds to RTX Pro 6000 #333

Description

@jameslamb

Description

RAPIDS projects generally need a GPU in their docs builds because things like Sphinx autosummary require actually import-ing the libraries, and many of the libraries require a GPU at runtime.

Today, projects are using the custom-job shared workflow and hard coding a GPU type, mostly L4.

  docs-build:
    uses: rapidsai/shared-workflows/.github/workflows/custom-job.yaml@main
    if: fromJSON(needs.changed-files.outputs.changed_file_groups).build_docs
    with:
      build_type: pull-request
      node_type: "gpu-l4-latest-1"
      arch: "amd64"
      container_image: "rapidsai/ci-conda:26.10-latest"
      script: "ci/build_docs.sh"

(example from NVIDIA/cuml)

L4 runners are heavily used in the shared CI runner system we use right now, which can result in long queue lines:

Look how often they've been 100% utilized in the last 7 days:

Image

We'd like to switch to a less-busy runner type, such as RTX Pro 6000:

Image

Benefits of this work

  • better utilization of shared CI resources
  • improved end-to-end CI time via less time stuck in "queued"

Acceptance Criteria

  • all RAPIDS projects using GPU runners to build docs in CI use RTX Pro 6000 runners

Approach

Centrallize that configuration in shared-workflows:

Test repos:

Remaining repos (GitHub search):

Notes

related: #322

Activity

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

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions