mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2026-10-25 22:28:17 +08:00
Apply consistent code formatting across the repo. Add clang-format and pre-commit hooks.
This commit is contained in:
Rob Armstrong
@@ -41,12 +41,13 @@
|
||||
#endif
|
||||
|
||||
/* Add two vectors on the GPU */
|
||||
__global__ void vectorAddGPU(float *a, float *b, float *c, int N) {
|
||||
int idx = blockIdx.x * blockDim.x + threadIdx.x;
|
||||
__global__ void vectorAddGPU(float *a, float *b, float *c, int N)
|
||||
{
|
||||
int idx = blockIdx.x * blockDim.x + threadIdx.x;
|
||||
|
||||
if (idx < N) {
|
||||
c[idx] = a[idx] + b[idx];
|
||||
}
|
||||
if (idx < N) {
|
||||
c[idx] = a[idx] + b[idx];
|
||||
}
|
||||
}
|
||||
|
||||
// Allocate generic memory with malloc() and pin it laster instead of using
|
||||
@@ -54,194 +55,196 @@ __global__ void vectorAddGPU(float *a, float *b, float *c, int N) {
|
||||
bool bPinGenericMemory = false;
|
||||
|
||||
// Macro to aligned up to the memory size in question
|
||||
#define MEMORY_ALIGNMENT 4096
|
||||
#define MEMORY_ALIGNMENT 4096
|
||||
#define ALIGN_UP(x, size) (((size_t)x + (size - 1)) & (~(size - 1)))
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
int n, nelem, deviceCount;
|
||||
int idev = 0; // use default device 0
|
||||
char *device = NULL;
|
||||
unsigned int flags;
|
||||
size_t bytes;
|
||||
float *a, *b, *c; // Pinned memory allocated on the CPU
|
||||
float *a_UA, *b_UA, *c_UA; // Non-4K Aligned Pinned memory on the CPU
|
||||
float *d_a, *d_b, *d_c; // Device pointers for mapped memory
|
||||
float errorNorm, refNorm, ref, diff;
|
||||
cudaDeviceProp deviceProp;
|
||||
int main(int argc, char **argv)
|
||||
{
|
||||
int n, nelem, deviceCount;
|
||||
int idev = 0; // use default device 0
|
||||
char *device = NULL;
|
||||
unsigned int flags;
|
||||
size_t bytes;
|
||||
float *a, *b, *c; // Pinned memory allocated on the CPU
|
||||
float *a_UA, *b_UA, *c_UA; // Non-4K Aligned Pinned memory on the CPU
|
||||
float *d_a, *d_b, *d_c; // Device pointers for mapped memory
|
||||
float errorNorm, refNorm, ref, diff;
|
||||
cudaDeviceProp deviceProp;
|
||||
|
||||
if (checkCmdLineFlag(argc, (const char **)argv, "help")) {
|
||||
printf("Usage: simpleZeroCopy [OPTION]\n\n");
|
||||
printf("Options:\n");
|
||||
printf(" --device=[device #] Specify the device to be used\n");
|
||||
printf(
|
||||
" --use_generic_memory (optional) use generic page-aligned for system "
|
||||
"memory\n");
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
|
||||
/* Get the device selected by the user or default to 0, and then set it. */
|
||||
if (getCmdLineArgumentString(argc, (const char **)argv, "device", &device)) {
|
||||
cudaGetDeviceCount(&deviceCount);
|
||||
idev = atoi(device);
|
||||
|
||||
if (idev >= deviceCount || idev < 0) {
|
||||
fprintf(stderr,
|
||||
"Device number %d is invalid, will use default CUDA device 0.\n",
|
||||
idev);
|
||||
idev = 0;
|
||||
if (checkCmdLineFlag(argc, (const char **)argv, "help")) {
|
||||
printf("Usage: simpleZeroCopy [OPTION]\n\n");
|
||||
printf("Options:\n");
|
||||
printf(" --device=[device #] Specify the device to be used\n");
|
||||
printf(" --use_generic_memory (optional) use generic page-aligned for system "
|
||||
"memory\n");
|
||||
return EXIT_SUCCESS;
|
||||
}
|
||||
}
|
||||
|
||||
// if GPU found supports SM 1.2, then continue, otherwise we exit
|
||||
if (!checkCudaCapabilities(1, 2)) {
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
/* Get the device selected by the user or default to 0, and then set it. */
|
||||
if (getCmdLineArgumentString(argc, (const char **)argv, "device", &device)) {
|
||||
cudaGetDeviceCount(&deviceCount);
|
||||
idev = atoi(device);
|
||||
|
||||
if (checkCmdLineFlag(argc, (const char **)argv, "use_generic_memory")) {
|
||||
if (idev >= deviceCount || idev < 0) {
|
||||
fprintf(stderr, "Device number %d is invalid, will use default CUDA device 0.\n", idev);
|
||||
idev = 0;
|
||||
}
|
||||
}
|
||||
|
||||
// if GPU found supports SM 1.2, then continue, otherwise we exit
|
||||
if (!checkCudaCapabilities(1, 2)) {
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
|
||||
if (checkCmdLineFlag(argc, (const char **)argv, "use_generic_memory")) {
|
||||
#if defined(__APPLE__) || defined(MACOSX)
|
||||
bPinGenericMemory = false; // Generic Pinning of System Paged memory is not
|
||||
// currently supported on Mac OSX
|
||||
bPinGenericMemory = false; // Generic Pinning of System Paged memory is not
|
||||
// currently supported on Mac OSX
|
||||
#else
|
||||
bPinGenericMemory = true;
|
||||
bPinGenericMemory = true;
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
if (bPinGenericMemory) {
|
||||
printf("> Using Generic System Paged Memory (malloc)\n");
|
||||
} else {
|
||||
printf("> Using CUDA Host Allocated (cudaHostAlloc)\n");
|
||||
}
|
||||
if (bPinGenericMemory) {
|
||||
printf("> Using Generic System Paged Memory (malloc)\n");
|
||||
}
|
||||
else {
|
||||
printf("> Using CUDA Host Allocated (cudaHostAlloc)\n");
|
||||
}
|
||||
|
||||
checkCudaErrors(cudaSetDevice(idev));
|
||||
checkCudaErrors(cudaSetDevice(idev));
|
||||
|
||||
/* Verify the selected device supports mapped memory and set the device
|
||||
flags for mapping host memory. */
|
||||
/* Verify the selected device supports mapped memory and set the device
|
||||
flags for mapping host memory. */
|
||||
|
||||
checkCudaErrors(cudaGetDeviceProperties(&deviceProp, idev));
|
||||
checkCudaErrors(cudaGetDeviceProperties(&deviceProp, idev));
|
||||
|
||||
#if CUDART_VERSION >= 2020
|
||||
|
||||
if (!deviceProp.canMapHostMemory) {
|
||||
fprintf(stderr, "Device %d does not support mapping CPU host memory!\n",
|
||||
idev);
|
||||
if (!deviceProp.canMapHostMemory) {
|
||||
fprintf(stderr, "Device %d does not support mapping CPU host memory!\n", idev);
|
||||
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
|
||||
checkCudaErrors(cudaSetDeviceFlags(cudaDeviceMapHost));
|
||||
#else
|
||||
fprintf(stderr,
|
||||
"CUDART version %d.%d does not support "
|
||||
"<cudaDeviceProp.canMapHostMemory> field\n",
|
||||
,
|
||||
CUDART_VERSION / 1000,
|
||||
(CUDART_VERSION % 100) / 10);
|
||||
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
|
||||
checkCudaErrors(cudaSetDeviceFlags(cudaDeviceMapHost));
|
||||
#else
|
||||
fprintf(stderr,
|
||||
"CUDART version %d.%d does not support "
|
||||
"<cudaDeviceProp.canMapHostMemory> field\n",
|
||||
, CUDART_VERSION / 1000, (CUDART_VERSION % 100) / 10);
|
||||
|
||||
exit(EXIT_SUCCESS);
|
||||
#endif
|
||||
|
||||
#if CUDART_VERSION < 4000
|
||||
|
||||
if (bPinGenericMemory) {
|
||||
fprintf(
|
||||
stderr,
|
||||
"CUDART version %d.%d does not support <cudaHostRegister> function\n",
|
||||
CUDART_VERSION / 1000, (CUDART_VERSION % 100) / 10);
|
||||
if (bPinGenericMemory) {
|
||||
fprintf(stderr,
|
||||
"CUDART version %d.%d does not support <cudaHostRegister> function\n",
|
||||
CUDART_VERSION / 1000,
|
||||
(CUDART_VERSION % 100) / 10);
|
||||
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
exit(EXIT_SUCCESS);
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
/* Allocate mapped CPU memory. */
|
||||
/* Allocate mapped CPU memory. */
|
||||
|
||||
nelem = 1048576;
|
||||
bytes = nelem * sizeof(float);
|
||||
nelem = 1048576;
|
||||
bytes = nelem * sizeof(float);
|
||||
|
||||
if (bPinGenericMemory) {
|
||||
if (bPinGenericMemory) {
|
||||
#if CUDART_VERSION >= 4000
|
||||
a_UA = (float *)malloc(bytes + MEMORY_ALIGNMENT);
|
||||
b_UA = (float *)malloc(bytes + MEMORY_ALIGNMENT);
|
||||
c_UA = (float *)malloc(bytes + MEMORY_ALIGNMENT);
|
||||
a_UA = (float *)malloc(bytes + MEMORY_ALIGNMENT);
|
||||
b_UA = (float *)malloc(bytes + MEMORY_ALIGNMENT);
|
||||
c_UA = (float *)malloc(bytes + MEMORY_ALIGNMENT);
|
||||
|
||||
// We need to ensure memory is aligned to 4K (so we will need to padd memory
|
||||
// accordingly)
|
||||
a = (float *)ALIGN_UP(a_UA, MEMORY_ALIGNMENT);
|
||||
b = (float *)ALIGN_UP(b_UA, MEMORY_ALIGNMENT);
|
||||
c = (float *)ALIGN_UP(c_UA, MEMORY_ALIGNMENT);
|
||||
// We need to ensure memory is aligned to 4K (so we will need to padd memory
|
||||
// accordingly)
|
||||
a = (float *)ALIGN_UP(a_UA, MEMORY_ALIGNMENT);
|
||||
b = (float *)ALIGN_UP(b_UA, MEMORY_ALIGNMENT);
|
||||
c = (float *)ALIGN_UP(c_UA, MEMORY_ALIGNMENT);
|
||||
|
||||
checkCudaErrors(cudaHostRegister(a, bytes, cudaHostRegisterMapped));
|
||||
checkCudaErrors(cudaHostRegister(b, bytes, cudaHostRegisterMapped));
|
||||
checkCudaErrors(cudaHostRegister(c, bytes, cudaHostRegisterMapped));
|
||||
checkCudaErrors(cudaHostRegister(a, bytes, cudaHostRegisterMapped));
|
||||
checkCudaErrors(cudaHostRegister(b, bytes, cudaHostRegisterMapped));
|
||||
checkCudaErrors(cudaHostRegister(c, bytes, cudaHostRegisterMapped));
|
||||
#endif
|
||||
} else {
|
||||
}
|
||||
else {
|
||||
#if CUDART_VERSION >= 2020
|
||||
flags = cudaHostAllocMapped;
|
||||
checkCudaErrors(cudaHostAlloc((void **)&a, bytes, flags));
|
||||
checkCudaErrors(cudaHostAlloc((void **)&b, bytes, flags));
|
||||
checkCudaErrors(cudaHostAlloc((void **)&c, bytes, flags));
|
||||
flags = cudaHostAllocMapped;
|
||||
checkCudaErrors(cudaHostAlloc((void **)&a, bytes, flags));
|
||||
checkCudaErrors(cudaHostAlloc((void **)&b, bytes, flags));
|
||||
checkCudaErrors(cudaHostAlloc((void **)&c, bytes, flags));
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
/* Initialize the vectors. */
|
||||
/* Initialize the vectors. */
|
||||
|
||||
for (n = 0; n < nelem; n++) {
|
||||
a[n] = rand() / (float)RAND_MAX;
|
||||
b[n] = rand() / (float)RAND_MAX;
|
||||
}
|
||||
for (n = 0; n < nelem; n++) {
|
||||
a[n] = rand() / (float)RAND_MAX;
|
||||
b[n] = rand() / (float)RAND_MAX;
|
||||
}
|
||||
|
||||
/* Get the device pointers for the pinned CPU memory mapped into the GPU
|
||||
memory space. */
|
||||
|
||||
#if CUDART_VERSION >= 2020
|
||||
checkCudaErrors(cudaHostGetDevicePointer((void **)&d_a, (void *)a, 0));
|
||||
checkCudaErrors(cudaHostGetDevicePointer((void **)&d_b, (void *)b, 0));
|
||||
checkCudaErrors(cudaHostGetDevicePointer((void **)&d_c, (void *)c, 0));
|
||||
checkCudaErrors(cudaHostGetDevicePointer((void **)&d_a, (void *)a, 0));
|
||||
checkCudaErrors(cudaHostGetDevicePointer((void **)&d_b, (void *)b, 0));
|
||||
checkCudaErrors(cudaHostGetDevicePointer((void **)&d_c, (void *)c, 0));
|
||||
#endif
|
||||
|
||||
/* Call the GPU kernel using the CPU pointers residing in CPU mapped memory.
|
||||
*/
|
||||
printf("> vectorAddGPU kernel will add vectors using mapped CPU memory...\n");
|
||||
dim3 block(256);
|
||||
dim3 grid((unsigned int)ceil(nelem / (float)block.x));
|
||||
vectorAddGPU<<<grid, block>>>(d_a, d_b, d_c, nelem);
|
||||
checkCudaErrors(cudaDeviceSynchronize());
|
||||
getLastCudaError("vectorAddGPU() execution failed");
|
||||
/* Call the GPU kernel using the CPU pointers residing in CPU mapped memory.
|
||||
*/
|
||||
printf("> vectorAddGPU kernel will add vectors using mapped CPU memory...\n");
|
||||
dim3 block(256);
|
||||
dim3 grid((unsigned int)ceil(nelem / (float)block.x));
|
||||
vectorAddGPU<<<grid, block>>>(d_a, d_b, d_c, nelem);
|
||||
checkCudaErrors(cudaDeviceSynchronize());
|
||||
getLastCudaError("vectorAddGPU() execution failed");
|
||||
|
||||
/* Compare the results */
|
||||
/* Compare the results */
|
||||
|
||||
printf("> Checking the results from vectorAddGPU() ...\n");
|
||||
errorNorm = 0.f;
|
||||
refNorm = 0.f;
|
||||
printf("> Checking the results from vectorAddGPU() ...\n");
|
||||
errorNorm = 0.f;
|
||||
refNorm = 0.f;
|
||||
|
||||
for (n = 0; n < nelem; n++) {
|
||||
ref = a[n] + b[n];
|
||||
diff = c[n] - ref;
|
||||
errorNorm += diff * diff;
|
||||
refNorm += ref * ref;
|
||||
}
|
||||
for (n = 0; n < nelem; n++) {
|
||||
ref = a[n] + b[n];
|
||||
diff = c[n] - ref;
|
||||
errorNorm += diff * diff;
|
||||
refNorm += ref * ref;
|
||||
}
|
||||
|
||||
errorNorm = (float)sqrt((double)errorNorm);
|
||||
refNorm = (float)sqrt((double)refNorm);
|
||||
errorNorm = (float)sqrt((double)errorNorm);
|
||||
refNorm = (float)sqrt((double)refNorm);
|
||||
|
||||
/* Memory clean up */
|
||||
/* Memory clean up */
|
||||
|
||||
printf("> Releasing CPU memory...\n");
|
||||
printf("> Releasing CPU memory...\n");
|
||||
|
||||
if (bPinGenericMemory) {
|
||||
if (bPinGenericMemory) {
|
||||
#if CUDART_VERSION >= 4000
|
||||
checkCudaErrors(cudaHostUnregister(a));
|
||||
checkCudaErrors(cudaHostUnregister(b));
|
||||
checkCudaErrors(cudaHostUnregister(c));
|
||||
free(a_UA);
|
||||
free(b_UA);
|
||||
free(c_UA);
|
||||
checkCudaErrors(cudaHostUnregister(a));
|
||||
checkCudaErrors(cudaHostUnregister(b));
|
||||
checkCudaErrors(cudaHostUnregister(c));
|
||||
free(a_UA);
|
||||
free(b_UA);
|
||||
free(c_UA);
|
||||
#endif
|
||||
} else {
|
||||
}
|
||||
else {
|
||||
#if CUDART_VERSION >= 2020
|
||||
checkCudaErrors(cudaFreeHost(a));
|
||||
checkCudaErrors(cudaFreeHost(b));
|
||||
checkCudaErrors(cudaFreeHost(c));
|
||||
checkCudaErrors(cudaFreeHost(a));
|
||||
checkCudaErrors(cudaFreeHost(b));
|
||||
checkCudaErrors(cudaFreeHost(c));
|
||||
#endif
|
||||
}
|
||||
}
|
||||
|
||||
exit(errorNorm / refNorm < 1.e-6f ? EXIT_SUCCESS : EXIT_FAILURE);
|
||||
exit(errorNorm / refNorm < 1.e-6f ? EXIT_SUCCESS : EXIT_FAILURE);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user