Skip to content

Clean up single-arch references and device arch arg in Rock manylinux.#473

Merged
tsrw2048 merged 1 commit into
rocm-jax-infrafrom
update-rock-manylinux-args
Jun 23, 2026
Merged

Clean up single-arch references and device arch arg in Rock manylinux.#473
tsrw2048 merged 1 commit into
rocm-jax-infrafrom
update-rock-manylinux-args

Conversation

@tsrw2048

Copy link
Copy Markdown
Contributor

Motivation

As per the discussion in #470, this PR does some more cleanup on TheRock manylinux image.

  • It refines the arch arg to take device-gfx<ARCH> instead of gfx<ARCH> so that a comma separated list of devices can be passed to TheRock pip install.

  • It cleans up old references to the single-arch Rock index.

Changes

All changes are contained to docker/manylinux/Dockerfile.jax-manylinux_2_28-therock

@tsrw2048 tsrw2048 merged commit b0a5652 into rocm-jax-infra Jun 23, 2026
4 of 13 checks passed
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