-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
main.cu only includes main project header
- Loading branch information
1 parent
f9e311d
commit 8d6548b
Showing
1 changed file
with
1 addition
and
50 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,50 +1 @@ | ||
#include <iostream> | ||
#include <math.h> | ||
|
||
// Kernel function to add the elements of two arrays | ||
__global__ void add(int n, float *x, float *y) | ||
{ | ||
|
||
int index = threadIdx.x + blockDim.x * blockIdx.x; | ||
int stride = blockDim.x * gridDim.x; | ||
|
||
for (int i = index; i < n; i += stride) | ||
y[i] = x[i] + y[i]; | ||
} | ||
|
||
int main(void) | ||
{ | ||
int N = 1 << 20; | ||
float *x, *y; | ||
|
||
// Allocate Unified Memory – accessible from CPU or GPU | ||
cudaMallocManaged(&x, N * sizeof(float)); | ||
cudaMallocManaged(&y, N * sizeof(float)); | ||
|
||
// initialize x and y arrays on the host | ||
for (int i = 0; i < N; i++) { | ||
x[i] = 1.0f; | ||
y[i] = 2.0f; | ||
} | ||
|
||
int block_size = 256; | ||
int num_blocks = (N + block_size - 1) / block_size; | ||
|
||
// Run kernel on 1M elements on the GPU | ||
add<<<num_blocks, block_size>>>(N, x, y); | ||
|
||
// Wait for GPU to finish before accessing on host | ||
cudaDeviceSynchronize(); | ||
|
||
// Check for errors (all values should be 3.0f) | ||
float maxError = 0.0f; | ||
for (int i = 0; i < N; i++) | ||
maxError = fmax(maxError, fabs(y[i] - 3.0f)); | ||
std::cout << "Max error: " << maxError << std::endl; | ||
|
||
// Free memory | ||
cudaFree(x); | ||
cudaFree(y); | ||
|
||
return 0; | ||
} | ||
#include <cuinterval/cuinterval.h> |