cuda programming example

Disgusted Duck answered on December 6, 2022 Popularity 10/10 Helpfulness 4/10

answer cuda programming example

cuda programming example

Comment

#include <stdio.h>

#define N 10

__global__ void add(int * a, int * b, int * c) 
{
    int tID = blockIdx.x;
    if (tID < N) 
    {
        c[tID] = a[tID] + b[tID];
    }
}
int main() 
{
    int a[N], b[N], c[N];
    int * dev_a, * dev_b, * dev_c;
    
    cudaMalloc((void ** ) & dev_a, N * sizeof(int));
    cudaMalloc((void ** ) & dev_b, N * sizeof(int));
    cudaMalloc((void ** ) & dev_c, N * sizeof(int));
    
    // Fill Arrays
    for (int i = 0; i < N; i++) 
    {
        a[i] = i,
        b[i] = 1;
    }
    
    cudaMemcpy(dev_a, a, N * sizeof(int), cudaMemcpyHostToDevice);
    cudaMemcpy(dev_b, b, N * sizeof(int), cudaMemcpyHostToDevice);
    
    add << < N, 1 >>> (dev_a, dev_b, dev_c);
    
    cudaMemcpy(c, dev_c, N * sizeof(int), cudaMemcpyDeviceToHost);
    
    for (int i = 0; i < N; i++) 
    {
        printf("%d + %d = %d\n", a[i], b[i], c[i]);
    }
    
    return 0;
    
}

Popularity 10/10 Helpfulness 4/10 Language c

Source: Grepper

Tags: c cuda

Link to this answer
Share Copy Link

Contributed on Dec 06 2022

Disgusted Duck

0 Answers Avg Quality 2/10

Continue with Google

By continuing, I agree that I have read and agree to Greppers's Terms of Service and Privacy Policy.

cuda programming example

Contents

More Related Answers

cuda programming example

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.