]> AND Private Git Repository - book_gpu.git/blobdiff - BookGPU/Chapters/chapter5/code/ex1.cu
Logo AND Algorithmique Numérique Distribuée

Private GIT Repository
last version
[book_gpu.git] / BookGPU / Chapters / chapter5 / code / ex1.cu
index 13d4322b84b9d86e351fc38996758718b298de97..847d20a6c73aa930b756ec462e4ea76ae044adbf 100644 (file)
@@ -1,29 +1,29 @@
-#include <gpulab/vector.h>\r
-\r
-__global__ void add(double* a, double const* b, int N)\r
-{\r
-       int i = blockDim.x*blockIdx.x + threadIdx.x;\r
-       if(i<N)\r
-               a[i] += b[i];\r
-}\r
-\r
-int main(int argc, char *argv[])\r
-{\r
-       int N = 1000;\r
-\r
-       // Plain CUDA example\r
-       double *a1, *b1;\r
-       cudaMalloc((void**)&a1, N*sizeof(double));\r
-       cudaMalloc((void**)&b1, N*sizeof(double));\r
-       cudaMemset(a1, 2.0, N);\r
-       cudaMemset(b1, 3.0, N);\r
-       int blocksize = 128;\r
-       add<<<(N+blocksize-1)/blocksize,blocksize>>>(a1, b1, N);\r
-\r
-       // gpulab example\r
-       gpulab::vector<double,gpulab::device_memory> a2(N, 2.0);\r
-       gpulab::vector<double,gpulab::device_memory> b2(N, 3.0);\r
-       a2.axpy(1.0, b2); // BLAS1: a2 = 1*b2 + a2\r
-       \r
-       return 0;\r
+#include <gpulab/vector.h>
+
+__global__ void add(double* a, double const* b, int N)
+{
+       int i = blockDim.x*blockIdx.x + threadIdx.x;
+       if(i<N)
+               a[i] += b[i];
+}
+
+int main(int argc, char *argv[])
+{
+       int N = 1000;
+
+       // Basic CUDA example
+       double *a1, *b1;
+       cudaMalloc((void**)&a1, N*sizeof(double));
+       cudaMalloc((void**)&b1, N*sizeof(double));
+       cudaMemset(a1, 2.0, N);
+       cudaMemset(b1, 3.0, N);
+       int blocksize = 128;
+       add<<<(N+blocksize-1)/blocksize,blocksize>>>(a1, b1, N);
+
+       // gpulab example
+       gpulab::vector<double,gpulab::device_memory> a2(N, 2.0);
+       gpulab::vector<double,gpulab::device_memory> b2(N, 3.0);
+       a2.axpy(1.0, b2); // BLAS1: a2 = 1*b2 + a2
+       
+       return 0;
 }
\ No newline at end of file
 }
\ No newline at end of file