Skip to content

Infer compute cap from CUDA_ARCH_LIST if possible #1401

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

Jump to bottom
Closed
wants to merge 1 commit into from
Closed

Infer compute cap from CUDA_ARCH_LIST if possible #1401

wants to merge 1 commit into from

Conversation

@michael132
Copy link
Contributor

@michael132 michael132 commented Apr 17, 2025

Enhance CUDA architecture support in build-jax.sh

  • Added logic to infer compute capabilities from the CUDA_ARCH_LIST variable if set.
  • Updated the supported_compute_capabilities function to handle both CUDA_ARCH_LIST and predefined architectures (amd64, arm64).

@gpupuck gpupuck requested review from gpupuck and olupton April 18, 2025 06:25
@michael132
Copy link
Contributor Author

michael132 commented Apr 21, 2025

Switched to a branch in the main repo to avoid CI issue
#1406

@michael132 michael132 closed this Apr 21, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@gpupuck gpupuck Awaiting requested review from gpupuck

@olupton olupton Awaiting requested review from olupton

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@michael132