add restrict for dst

This commit is contained in:
pidack 2024-08-29 10:36:33 +08:00
parent 5999d6d06e
commit 316a049533
2 changed files with 2 additions and 2 deletions

View file

@ -5,7 +5,7 @@ static __global__ void ssm_conv_f32(
const float * __restrict__ src0, const float * __restrict__ src1, const float * __restrict__ src0, const float * __restrict__ src1,
const int src0_nb0, const int src0_nb1, const int src0_nb2, const int src0_nb0, const int src0_nb1, const int src0_nb2,
const int src1_nb1, const int src1_nb1,
float * dst, float * __restrict__ dst,
const int dst_nb0, const int dst_nb1, const int dst_nb2, const int dst_nb0, const int dst_nb1, const int dst_nb2,
const int nc, const int ncs, const int nr, const int n_t, const int n_s) { const int nc, const int ncs, const int nr, const int n_t, const int n_s) {

View file

@ -10,7 +10,7 @@ static __global__ void ssm_scan_f32(
const int src3_nb1, const int src3_nb1,
const int src4_nb1, const int src4_nb2, const int src4_nb1, const int src4_nb2,
const int src5_nb1, const int src5_nb2, const int src5_nb1, const int src5_nb2,
float * dst, float * __restrict__ dst,
const int nc, const int nr, const int n_t, const int n_s) { const int nc, const int nr, const int n_t, const int n_s) {
// const int row = blockIdx.x*blockDim.y + threadIdx.y; // const int row = blockIdx.x*blockDim.y + threadIdx.y;