2024-07-12
한어Русский языкEnglishFrançaisIndonesianSanskrit日本語DeutschPortuguêsΕλληνικάespañolItalianoSuomalainenLatina
#include <stdio.h>
// includes CUDA Runtime
#include <cuda_runtime.h>
#include <cuda_profiler_api.h>
// includes, project
#include <helper_cuda.h>
#include <helper_functions.h> // helper utility functions
__global__ void increment_kernel(int *g_data, int inc_value) {
int idx = blockIdx.x * blockDim.x + threadIdx.x;
g_data[idx] = g_data[idx] + inc_value;
}
bool correct_output(int *data, const int n, const int x) {
for (int i = 0; i < n; i++)
if (data[i] != x) {
printf("Error! data[%d] = %d, ref = %dn", i, data[i], x);
return false;
}
return true;
}
int main(int argc, char *argv[]) {
int devID;
cudaDeviceProp deviceProps;
printf("[%s] - Starting...n", argv[0]);
// This will pick the best possible CUDA capable device
devID = findCudaDevice(argc, (const char **)argv);
// get device name
checkCudaErrors(cudaGetDeviceProperties(&deviceProps, devID));
printf("CUDA device [%s]n", deviceProps.name);
int n = 16 * 1024 * 1024;
int nbytes = n * sizeof(int);
int value = 26;
// allocate host memory
int *a = 0;
checkCudaErrors(cudaMallocHost((void **)&a, nbytes));
memset(a, 0, nbytes);
// allocate device memory
int *d_a = 0;
checkCudaErrors(cudaMalloc((void **)&d_a, nbytes));
checkCudaErrors(cudaMemset(d_a, 255, nbytes));
// set kernel launch configuration
dim3 threads = dim3(512, 1);
dim3 blocks = dim3(n / threads.x, 1);
// create cuda event handles
cudaEvent_t start, stop;
checkCudaErrors(cudaEventCreate(&start));
checkCudaErrors(cudaEventCreate(&stop));
StopWatchInterface *timer = NULL;
sdkCreateTimer(&timer);
sdkResetTimer(&timer);
checkCudaErrors(cudaDeviceSynchronize());
float gpu_time = 0.0f;
// asynchronously issue work to the GPU (all to stream 0)
checkCudaErrors(cudaProfilerStart());
sdkStartTimer(&timer);
cudaEventRecord(start, 0);
cudaMemcpyAsync(d_a, a, nbytes, cudaMemcpyHostToDevice, 0);
increment_kernel<<<blocks, threads, 0, 0>>>(d_a, value);
cudaMemcpyAsync(a, d_a, nbytes, cudaMemcpyDeviceToHost, 0);
cudaEventRecord(stop, 0);
sdkStopTimer(&timer);
checkCudaErrors(cudaProfilerStop());
// have CPU do some work while waiting for stage 1 to finish
unsigned long int counter = 0;
while (cudaEventQuery(stop) == cudaErrorNotReady) {
counter++;
}
checkCudaErrors(cudaEventElapsedTime(&gpu_time, start, stop));
// print the cpu and gpu times
printf("time spent executing by the GPU: %.2fn", gpu_time);
printf("time spent by CPU in CUDA calls: %.2fn", sdkGetTimerValue(&timer));
printf("CPU executed %lu iterations while waiting for GPU to finishn",
counter);
// check the output for correctness
bool bFinalResults = correct_output(a, n, value);
// release resources
checkCudaErrors(cudaEventDestroy(start));
checkCudaErrors(cudaEventDestroy(stop));
checkCudaErrors(cudaFreeHost(a));
checkCudaErrors(cudaFree(d_a));
exit(bFinalResults ? EXIT_SUCCESS : EXIT_FAILURE);
}
AusrüstungInitialisierung:Die Funktion findCudaDevice wird verwendet, um das beste CUDA-Gerät auszuwählen und die Geräte-ID zurückzugeben.
devID = findCudaDevice(argc, (const char **)argv);
Geräteeigenschaften abrufen: Die Funktion cudaGetDeviceProperties ruft die Eigenschaften des angegebenen Geräts ab, einschließlich Gerätename und anderen Informationen.
checkCudaErrors(cudaGetDeviceProperties(&deviceProps, devID));
Speicherzuweisung: Verwenden Sie cudaMallocHost, um seitengesperrten Speicher zuzuweisen, auf den auf der CPU zugegriffen werden kann, und cudaMalloc, um Speicher auf dem Gerät zuzuweisen.
int *a = 0;
checkCudaErrors(cudaMallocHost((void **)&a, nbytes));
Legen Sie den Thread-Block und das Raster fest: Hier wird die Thread-Blockgröße auf 512 Threads festgelegt und die Rastergröße wird dynamisch basierend auf der Datengröße berechnet.
dim3 threads = dim3(512, 1);
dim3 blocks = dim3(n / threads.x, 1);
Erstellen Sie CUDA-Ereignisse und Timer: CUDA-Ereignisse werden zum Aufzeichnen der Zeit und Timer zum Messen der CPU-Ausführungszeit verwendet.
cudaEvent_t start, stop;
checkCudaErrors(cudaEventCreate(&start));
checkCudaErrors(cudaEventCreate(&stop));
CUDA-Stream-Verarbeitung: Verwendung von cudaMemcpyAsync für asynchrone Speicherkopie, <<<blocks, threads> >>Syntax startet die gleichzeitig ausgeführte CUDA-Kernelfunktion increment_kernel.
cudaMemcpyAsync(d_a, a, nbytes, cudaMemcpyHostToDevice, 0);
increment_kernel<<<blocks, threads, 0, 0>>>(d_a, value);
cudaMemcpyAsync(a, d_a, nbytes, cudaMemcpyDeviceToHost, 0);
Timing und Warten: cudaEventRecord zeichnet Ereignisse auf und wird zur Berechnung der GPU-Ausführungszeit verwendet. Warten Sie, bis der GPU-Vorgang über cudaEventQuery(stop) abgeschlossen ist.
cudaEventRecord(start, 0);
// ...
cudaEventRecord(stop, 0);
Ergebnisüberprüfung: Verwenden Sie die Funktion „correct_output“, um die Richtigkeit der GPU-Berechnungsergebnisse zu überprüfen.
bool bFinalResults = correct_output(a, n, value);
Ressourcenfreigabe: Geben Sie zugewiesenen Speicher und CUDA-Ereignisse frei.
checkCudaErrors(cudaEventDestroy(start));
checkCudaErrors(cudaEventDestroy(stop));
checkCudaErrors(cudaFreeHost(a));
checkCudaErrors(cudaFree(d_a));
CUDA-Kernel-Funktion increment_kernel:
Diese einfache CUDA-Kernelfunktion wird verwendet, um jedes Element in einem Array um einen angegebenen Wert inc_value zu erhöhen. blockIdx.x und threadIdx.x werden verwendet, um die globale Index-IDX jedes Threads zu berechnen und dann die Additionsoperation auszuführen.
__global__ void increment_kernel(int *g_data, int inc_value) {
int idx = blockIdx.x * blockDim.x + threadIdx.x;
g_data[idx] = g_data[idx] + inc_value;
}
Weitere Hilfsfunktionen
checkCudaErrors:Überprüfen Sie CUDAOb beim Funktionsaufruf ein Fehler vorliegt.
sdkCreateTimer und sdkResetTimer: werden zum Erstellen und Zurücksetzen von Timern verwendet.
sdkStartTimer und sdkStopTimer: werden zum Starten und Stoppen von Timern und zum Aufzeichnen der CPU-Ausführungszeit verwendet.