Skip to content

[ORT] Auto-detection for CUDA and ROCM path#6093

Merged
davidrohr merged 12 commits intoalisw:masterfrom
ChSonnabend:autodetect_cudarocm_path
Jan 23, 2026
Merged

[ORT] Auto-detection for CUDA and ROCM path#6093
davidrohr merged 12 commits intoalisw:masterfrom
ChSonnabend:autodetect_cudarocm_path

Conversation

@ChSonnabend
Copy link
Copy Markdown
Collaborator

No description provided.

@ChSonnabend ChSonnabend requested a review from a team as a code owner January 22, 2026 12:19
Copy link
Copy Markdown
Contributor

@davidrohr davidrohr left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, nice, this is in principle what I meant, but I think we need to still improve some things:

Comment thread gpu-system.sh Outdated
Comment thread gpu-system.sh Outdated
Comment thread gpu-system.sh Outdated
Comment thread gpu-system.sh Outdated
Comment thread gpu-system.sh
if [[ "${PKG_VERSION}" =~ (^|-)rocm_arch@ ]]; then

### ROCm
if [[ "${O2_GPU_ROCM_AVAILABLE}" == "1" && "${PKG_VERSION}" =~ rocm_home@([^@]*)@ ]]; then
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

O2_GPU_ROCM_AVAILABLE can be grouped, no?

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It could, but then I think the else statements have to cover more cases

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Comment by David after I originally did the grouping: "I don't understand why you restructure it so much, can't you just add another block". So I kept the block structure. I don't care which way we go.

ChSonnabend and others added 2 commits January 22, 2026 20:43
Co-authored-by: Giulio Eulisse <10544+ktf@users.noreply.github.com>
@davidrohr davidrohr merged commit b4e60eb into alisw:master Jan 23, 2026
13 of 14 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.

3 participants