Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- int *a, *b, *c;
- int *d_a, *d_b, *d_c;
- a = (int *)malloc(SIZE*sizeof(int));
- b = (int *)malloc(SIZE*sizeof(int));
- c = (int *)malloc(SIZE*sizeof(int));
- cudaMalloc( &d_a, SIZE*sizeof(int));
- cudaMalloc( &d_b, SIZE*sizeof(int));
- cudaMalloc( &d_c, SIZE*sizeof(int));
- for( int i = 0; i < SIZE; i++ )
- {
- a[i] =i;
- b[i] =i;
- c[i] =0;
- }
- cudaMemcpy( d_a, a, SIZE*sizeof(int), cudaMemcpyHostToDevice );
- cudaMemcpy( d_b, b, SIZE*sizeof(int), cudaMemcpyHostToDevice );
- cudaMemcpy( d_c, c, SIZE*sizeof(int), cudaMemcpyHostToDevice );
- InitialAdd<<< 3 , SIZE >>>( d_a, d_b, d_c, SIZE);
- cudaMemcpy( c, d_c, SIZE*sizeof(int), cudaMemcpyDeviceToHost );
- for( int i = 0; i < 5000; i++)
- printf("c[%d] = %dn", i, c[i]);
- free(a);
- free(b);
- free(c);
- cudaFree(d_a);
- cudaFree(d_b);
- cudaFree(d_c);
- return 0;
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement