This commit is contained in:
陈六生 2024-08-01 15:57:48 +08:00 committed by GitHub
commit 55f5ace42c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -158,11 +158,13 @@ int main(int argc, char **argv) {
// Grid/Block configuration
int threadsPerBlock = 256;
int blocksPerGrid = (N + threadsPerBlock - 1) / threadsPerBlock;
void *kernel_launch_config[5] = {CU_LAUNCH_PARAM_BUFFER_POINTER,
argBuffer, CU_LAUNCH_PARAM_BUFFER_SIZE,
&offset, CU_LAUNCH_PARAM_END};
// Launch the CUDA kernel
checkCudaErrors(cuLaunchKernel(vecAdd_kernel, blocksPerGrid, 1, 1,
threadsPerBlock, 1, 1, 0, NULL, NULL,
argBuffer));
kernel_launch_config));
}
#ifdef _DEBUG