Skip to content
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

JAX: Support CUDA 12 #223

Closed
jakirkham opened this issue Jan 25, 2024 · 3 comments
Closed

JAX: Support CUDA 12 #223

jakirkham opened this issue Jan 25, 2024 · 3 comments

Comments

@jakirkham
Copy link
Member

Filing to track CUDA 12 bringup on JAX

Related issues:
@jakirkham jakirkham mentioned this issue Jan 25, 2024
Closed
@dhirschfeld
Copy link
Member

My current understanding is that the GitHub Runners aren't powerful enough to build the entire stack:

The job running on agent Azure Pipelines 31 ran longer than the maximum time of 360 minutes.

...and the failure of a few prevents any versions from being published:

Only 4 out of 21 failed.

image

#207 (comment)

I'm not sure if there has been any movement since, but I'm certainly interested in CUDA 12 support (running in ubuntu wsl) 👀

@jakirkham
Copy link
Member Author

Some feedstocks have started using Quansight CI. AIUI it involves

There may be other steps I'm forgetting, but that is at least a good starting point

@weiji14 weiji14 mentioned this issue Feb 6, 2024
Closed
@ngam
Copy link
Contributor

ngam commented Mar 17, 2024

Done in #227

@ngam ngam closed this as completed Mar 17, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

Rebuild for CUDA 12 w/arch + Windows support regro-cf-autotick-bot/jaxlib-feedstock
3 participants
@dhirschfeld @jakirkham @ngam