Skip to content

To check rocm py_torch ci build #4761

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.

Sign up for GitHub

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

Open
renjithravindrankannath wants to merge 6 commits into
base: develop
Choose a base branch
from
Open

To check rocm py_torch ci build #4761

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
aa1864c
To check rocm py_torch ci build
renjithravindrankannath May 8, 2026
dd71e8d
removing unwanted $ in the spec string
renjithravindrankannath May 11, 2026
d7710b7
exclude is not required
renjithravindrankannath May 11, 2026
55cfb81
Updating aotriton-llvm version dependency
renjithravindrankannath May 12, 2026
d9a7c5c
hwloc with older rocm versions conflicts with llvm@20 while linking
renjithravindrankannath May 12, 2026
e9d44d8
Merge branch 'develop' into py-torch-ci-build
Jun 17, 2026
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
2 changes: 2 additions & 0 deletions .ci/gitlab/configs/linux/ci.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,8 @@ ci:
- wrf
build-job:
tags: [ "spack", "huge" ]
# 24h for long ROCm/ML rebuilds; GitLab project/runner max must allow this
timeout: 1440 minutes
variables:
CI_JOB_SIZE: huge
SPACK_BUILD_JOBS: "12"
Expand Down
2 changes: 1 addition & 1 deletion repos/spack_repo/builtin/packages/aotriton/package.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ class Aotriton(CMakePackage):

# build llvm version with mlir with the commit that matches inside the llvm-hash.txt
depends_on("aotriton-llvm@0.10", when="@0.10b:")
depends_on("aotriton-llvm@0.9", when="@0.9b")
depends_on("aotriton-llvm@0.9", when="@0.9b:0.9.2b")
depends_on("aotriton-llvm@0.8", when="@0.8b")

conflicts("^openssl@3.3.0")
Expand Down
4 changes: 4 additions & 0 deletions repos/spack_repo/builtin/packages/py_llvmlite/package.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,10 @@ class PyLlvmlite(PythonPackage):
depends_on("llvm@14", when="@0.41:0.43")
depends_on("llvm@11:14", when="@0.40")
depends_on("llvm@11", when="@0.37:0.39")
# When the spec has both llvm@20 and hwloc+rocm (e.g. llvm build-depends on hwloc),
# hwloc with older rocm versions conflicts with llvm@20 while linking.
# Hence adding hwloc~rocm dependency for llvmlite@0.45:.
depends_on("hwloc ~rocm", when="@0.45:")
for t in [
"arm:",
"ppc:",
Expand Down
11 changes: 4 additions & 7 deletions stacks/ml-linux-x86_64/spack.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -113,15 +113,11 @@ spack:
- $jax_specs
- matrix:
- [$keras_specs]
exclude:
- py-keras backend=torch
# Does not yet support Spack-installed ROCm
# - $torch_specs
- $torch_specs
- $scikit_learn_specs
- $tensorboard_specs
- $tensorflow_specs
# Does not yet support Spack-installed ROCm
# - $xgboost_specs
- $xgboost_specs
override:
packages:
all:
Expand All @@ -139,8 +135,9 @@ spack:
- target=x86_64_v3
- ~cuda
- +rocm
- ~distributed
- amdgpu_target=gfx90a
- ~flash_attention
- +flash_attention
- "%c,cxx=gcc"

ci:
Expand Down
Loading