NVIDIA / NVIDIA/CUDALibrarySamples
cuBLASDx samples feedback
@llukas is already working on this.
Since Jun 11, 2025.
- Dominant language
- Cuda
- Stars
- 2.5k
- Forks
- 478
- PR merge metrics
- No merged PRs in 30d
Description
Hello I wanted to see if cuBLASDx was a feasible replacement for something I thought I needed to do with cutlass/cute. Here was my workflow:
- Oh they have a fused kernel example!
- Oh, it's only launching one block?
- Oh, they have a multiblock example (and only one besides attn and fft), but it seems slow. Let's try boosting up K.
- Static assertion for block size kicks in. I read comment saying how it doesn't split on K. I change 'K' and specify 'k' as 1024 and 64 respectively.
- A cooperative copy assertion kicks in that I don't understand.
I'm curious about these choices of making almost all of the examples a launch with grid_size=1. If the kernels were mostly complete it would be decently easy to fiddle with and figure out how to do something specific to your needs. It also seems like it would be good to at least have one canonical example that just implements K as arbitrary for a simple gemm (maybe command line params to get a feel for the speed?)
FWIW i'm trying to do an outer product of two matrices and then issue an fma on the result in place against another matrix in vram. A fused kernel for this would save an unbelievable amount of vram traffic in certain problem sizes.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.