add restrict for dst
This commit is contained in:
parent
5999d6d06e
commit
316a049533
2 changed files with 2 additions and 2 deletions
|
@ -5,7 +5,7 @@ static __global__ void ssm_conv_f32(
|
|||
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,
|
||||
float * __restrict__ dst,
|
||||
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) {
|
||||
|
||||
|
|
|
@ -10,7 +10,7 @@ static __global__ void ssm_scan_f32(
|
|||
const int src3_nb1,
|
||||
const int src4_nb1, const int src4_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 row = blockIdx.x*blockDim.y + threadIdx.y;
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue