Fix ifdef and old code

nond-rep
sfilippone 12 months ago
parent 35d68aa4e3
commit 0568a83734

@ -78,6 +78,7 @@ void spgpuCaxpby(spgpuHandle_t handle,
#else #else
__global__ void spgpuCaxpby_krn(cuFloatComplex *z, int n, cuFloatComplex beta, cuFloatComplex *y, cuFloatComplex alpha, cuFloatComplex* x) __global__ void spgpuCaxpby_krn(cuFloatComplex *z, int n, cuFloatComplex beta, cuFloatComplex *y, cuFloatComplex alpha, cuFloatComplex* x)
{ {
int id = threadIdx.x + BLOCK_SIZE*blockIdx.x; int id = threadIdx.x + BLOCK_SIZE*blockIdx.x;
@ -94,6 +95,7 @@ __global__ void spgpuCaxpby_krn(cuFloatComplex *z, int n, cuFloatComplex beta, c
} }
} }
void spgpuCaxpby_(spgpuHandle_t handle, void spgpuCaxpby_(spgpuHandle_t handle,
__device cuFloatComplex *z, __device cuFloatComplex *z,
int n, int n,
@ -103,15 +105,9 @@ void spgpuCaxpby_(spgpuHandle_t handle,
__device cuFloatComplex* x) __device cuFloatComplex* x)
{ {
int msize = (n+BLOCK_SIZE-1)/BLOCK_SIZE; int msize = (n+BLOCK_SIZE-1)/BLOCK_SIZE;
int num_mp, max_threads_mp, num_blocks_mp, num_blocks;
dim3 block(BLOCK_SIZE); dim3 block(BLOCK_SIZE);
cudaDeviceProp deviceProp; dim3 grid(msize);
cudaGetDeviceProperties(&deviceProp, 0);
num_mp = deviceProp.multiProcessorCount;
max_threads_mp = deviceProp.maxThreadsPerMultiProcessor;
num_blocks_mp = max_threads_mp/BLOCK_SIZE;
num_blocks = num_blocks_mp*num_mp;
dim3 grid(num_blocks);
spgpuCaxpby_krn<<<grid, block, 0, handle->currentStream>>>(z, n, beta, y, alpha, x); spgpuCaxpby_krn<<<grid, block, 0, handle->currentStream>>>(z, n, beta, y, alpha, x);
} }

@ -89,6 +89,7 @@ __global__ void spgpuDaxpby_krn(double *z, int n, double beta, double *y, double
} }
} }
void spgpuDaxpby_(spgpuHandle_t handle, void spgpuDaxpby_(spgpuHandle_t handle,
__device double *z, __device double *z,
int n, int n,
@ -98,15 +99,9 @@ void spgpuDaxpby_(spgpuHandle_t handle,
__device double* x) __device double* x)
{ {
int msize = (n+BLOCK_SIZE-1)/BLOCK_SIZE; int msize = (n+BLOCK_SIZE-1)/BLOCK_SIZE;
int num_mp, max_threads_mp, num_blocks_mp, num_blocks;
dim3 block(BLOCK_SIZE); dim3 block(BLOCK_SIZE);
cudaDeviceProp deviceProp; dim3 grid(msize);
cudaGetDeviceProperties(&deviceProp, 0);
num_mp = deviceProp.multiProcessorCount;
max_threads_mp = deviceProp.maxThreadsPerMultiProcessor;
num_blocks_mp = max_threads_mp/BLOCK_SIZE;
num_blocks = num_blocks_mp*num_mp;
dim3 grid(num_blocks);
spgpuDaxpby_krn<<<grid, block, 0, handle->currentStream>>>(z, n, beta, y, alpha, x); spgpuDaxpby_krn<<<grid, block, 0, handle->currentStream>>>(z, n, beta, y, alpha, x);
} }
@ -134,6 +129,7 @@ void spgpuDaxpby(spgpuHandle_t handle,
cudaCheckError("CUDA error on daxpby"); cudaCheckError("CUDA error on daxpby");
} }
#endif #endif
void spgpuDmaxpby(spgpuHandle_t handle, void spgpuDmaxpby(spgpuHandle_t handle,
__device double *z, __device double *z,

Loading…
Cancel
Save