From c6047a0db53014ccfdb0fa40ff7e2d55473b6fb1 Mon Sep 17 00:00:00 2001 From: JohannesGaessler Date: Sun, 31 Dec 2023 16:37:26 +0100 Subject: [PATCH] 4 cuda streams --- ggml-cuda.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ggml-cuda.cu b/ggml-cuda.cu index 15729691a..dc3237a45 100644 --- a/ggml-cuda.cu +++ b/ggml-cuda.cu @@ -522,7 +522,7 @@ static_assert(K_QUANTS_PER_ITERATION == 1 || K_QUANTS_PER_ITERATION == 2, "K_QUA #define MUL_MAT_SRC1_COL_STRIDE 128 -#define MAX_STREAMS 8 +#define MAX_STREAMS 4 static cudaStream_t g_cudaStreams[GGML_CUDA_MAX_DEVICES][MAX_STREAMS] = { { nullptr } }; struct ggml_tensor_extra_gpu {