hipBLASLt icon indicating copy to clipboard operation
hipBLASLt copied to clipboard

Add support for single wave stream-k kernels

Open AlexBrownAMD opened this issue 8 months ago • 0 comments

Allow wavegroup to be less than 4 in stream-k kernels. This change updates the partials and fixup code to tkae number of waves into account.

Added new test cases to the hgemm stream-k unit tests for wave groups (2,1), (1,2), and (1,1).

AlexBrownAMD avatar Apr 23 '25 15:04 AlexBrownAMD