Skip to content

Cublas Support#2444

Draft
dsding2 wants to merge 8 commits intoEnzymeAD:mainfrom
kevin314:squash
Draft

Cublas Support#2444
dsding2 wants to merge 8 commits intoEnzymeAD:mainfrom
kevin314:squash

Conversation

@dsding2
Copy link
Copy Markdown

@dsding2 dsding2 commented Apr 22, 2026

Adds support for cublasSgemm, including caching and runtime shape propagation.

@dsding2 dsding2 changed the title Squash Cublas Support Apr 22, 2026
@dsding2 dsding2 marked this pull request as draft April 22, 2026 17:54
@mofeing
Copy link
Copy Markdown
Collaborator

mofeing commented Apr 22, 2026

Doesn't XLA already lower stablehlo.dot_general to cuBlas<t>gemm when running on GPU?

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