Thursday, February 16, 2012

My First CUDA Program

This is my first CUDA program.



#include <stdio.h>
#include <cuda_runtime.h>


bool InitCUDA()
{
    int count;


    cudaGetDeviceCount(&count);
    if(count == 0) {
        fprintf(stderr, "There is no device.\n");
        return false;
    }


    int i;
    for(i = 0; i < count; i++) {
        cudaDeviceProp prop;
        if(cudaGetDeviceProperties(&prop, i) == cudaSuccess) {
            if(prop.major >= 1) {
                break;
            }
        }
    }


    if(i == count) {
        fprintf(stderr, "There is no device supporting CUDA 1.x.\n");
        return false;
    }


    cudaSetDevice(i);


    return true;
}


int main()
{
    if(!InitCUDA()) {
        return 0;
    }


    printf("CUDA initialized.\n");


    return 0;
}

No comments:

Post a Comment