From 0e682ced5ed35665c31bec612f3e6e3d9c4abf64 Mon Sep 17 00:00:00 2001 From: pidack Date: Tue, 27 Aug 2024 20:54:39 +0800 Subject: [PATCH] add restrict --- ggml/src/ggml-cuda/ssm_conv.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ggml/src/ggml-cuda/ssm_conv.cu b/ggml/src/ggml-cuda/ssm_conv.cu index 0e51bf771..ec25376c5 100644 --- a/ggml/src/ggml-cuda/ssm_conv.cu +++ b/ggml/src/ggml-cuda/ssm_conv.cu @@ -2,7 +2,7 @@ template static __global__ void ssm_conv_f32( - const float * src0, const float * src1, + const float * __restrict__ src0, const float * __restrict__ src1, const int src0_nb0, const int src0_nb1, const int src0_nb2, const int src1_nb1, float * dst,