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

Add wrappers for gemv_batched! #1981

Merged
merged 8 commits into from
Jul 18, 2023
Merged

Add wrappers for gemv_batched! #1981

merged 8 commits into from
Jul 18, 2023

Conversation

lpawela
Copy link
Contributor

@lpawela lpawela commented Jun 27, 2023

This adds wrappers and updates @ccall arguments for various versions of batched matrix vector multiplication.

@lpawela lpawela marked this pull request as ready for review June 28, 2023 00:58
maleadt
maleadt reviewed Jun 29, 2023
View reviewed changes
lib/cublas/wrappers.jl Outdated Show resolved Hide resolved
@maleadt maleadt added enhancement New feature or request cuda libraries Stuff about CUDA library wrappers. labels Jun 29, 2023
@maleadt
Copy link
Member

maleadt commented Jul 12, 2023

Looks like cublasSgemvBatched is only available on CUDA 11.6+, so you'll have to gate the tests accordingly. Maybe gate on the CUBLAS version; CUDA 11.6 ships CUBLAS 11.9.2 so if CUBLAS.version() >= v"11.9" or so.

@maleadt maleadt merged commit cb6ebaa into JuliaGPU:master Jul 18, 2023
1 check passed
@maleadt
Copy link
Member

maleadt commented Jul 18, 2023

Thanks!

@lpawela lpawela deleted the lp/gemv-batched branch March 20, 2024 13:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@maleadt maleadt maleadt left review comments

Assignees
No one assigned
Labels
cuda libraries Stuff about CUDA library wrappers. enhancement New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
2 participants
@lpawela @maleadt