NVIDIA / NVIDIA/cutlass

[QST] How to do concurrent GEMMs ?

Open
#1,418 8 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

inactive-30d inactive-90d question
Dominant language
C++
Stars
10.5k
Forks
2.1k
Avg merge
3d 11h
Merged PRs (30d)
7

Description

Hello,
Is it possible to launch concurrent GEMMs from the host? (only using more CPU threads as a last resort) I have used streams, but they are not running concurrently, but sequentially (which was what I expected from the code). Is there a way to do it or is there a more efficient way using another kind of GEMM instead of the basic template? Thank you, below is the code I am using to call the GEMMs.

for(int i = 0; i < 4; i++ ){
      for(int j = 0; j < 4; j++){
        CutlassGemm(m,n,k,W1[i], m, Input_8bit_pointer[j], m,Interm1[i*4+j], m, cuda_streams[i][j], i, j );
      }
}

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start from the CutlassGemm call and the cuda_streams[i][j] setup shown in the issue. Trace how the four-by-four GEMM launches are configured and determine whether the reported sequential execution is expected; document the supported path or limitation and verify it with a focused concurrency test.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp
Domain
performance
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.