[BUG] example 06_splitK_gemm should allow post-Volta GPUs
Nobody has claimed this yet.
- Dominant language
- C++
- Stars
- 10.5k
- Forks
- 2.1k
- Avg merge
- 3d 11h
- Merged PRs (30d)
- 7
Description
Describe the bug
Only Volta GPUs are allowed to run the example.
https://github.com/NVIDIA/cutlass/blob/e1976daacc7b030ba672217eb5d96f5a663df4ab/examples/06_splitK_gemm/splitk_gemm.cu#L188-L189
Steps/Code to reproduce bug
Run the compiled program on a machine with compute capability >= 80.
Expected behavior
The program should work on Volta and later GPUs.
Additional context
I have locally tested that this could work on 89.
Contributor guide
No contributing guide indexed for this repository
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.
Research direction
Start in examples/06_splitK_gemm/splitk_gemm.cu at lines 188-189 and inspect the GPU capability check. Run the compiled example on a Volta GPU and on a machine with compute capability >= 80, including the reported 89 case. Done means the example works on Volta and later GPUs.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- cpp
- Domain
- hpc
- Issue type
- Bug
- Difficulty
- 2/5
- Estimated time
- 1-3 hours
- Activity status
- Stale
- Clarity
- Clearly specified
- Newbie friendliness
- 65/100