tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-08-18 05:56:00 -04:00
Code Issues Packages Projects Releases Wiki Activity

CUDA: stream-k decomposition for MMQ (#8018)

Browse Source 
* CUDA: stream-k decomposition for MMQ

* fix undefined memory reads for small matrices
This commit is contained in:
Johannes Gäßler
2024-06-20 14:39:21 +02:00
committed by GitHub
parent 2075a66a96
commit d50f8897a7
4 changed files with 292 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ggml-cuda/common.cuh
View File

@@ -652,8 +652,8 @@ static int get_mmq_x_max_host(const int cc) {
}
// Round rows to this value for --split-mode row:
static int get_mmq_y_host(const int cc, const int mmq_x) {
return cc >= CC_VOLTA && mmq_x >= 32 ? 128 : 64;
static int get_mmq_y_host(const int cc) {
return cc >= CC_VOLTA ? 128 : 64;
}
//////////////////////