Function main

cpp/0_Introduction/vectorAddMMAP/vectorAddMMAP.cpp:116–228

Host code

// Host code
int main(int argc, char **argv)
{
    printf("Vector Addition (Driver API)\n");
    int    N            = 50000;
    size_t size         = N * sizeof(float);
    int    attributeVal = 0;

    // Initialize
    checkCudaErrors(cuInit(0));

    cuDevice = findCudaDeviceDRV(argc, (const char **)argv);

    // Check that the selected device supports virtual address management
    checkCudaErrors(
        cuDeviceGetAttribute(&attributeVal, CU_DEVICE_ATTRIBUTE_VIRTUAL_ADDRESS_MANAGEMENT_SUPPORTED, cuDevice));
    printf("Device %d VIRTUAL ADDRESS MANAGEMENT SUPPORTED = %d.\n", cuDevice, attributeVal);
    if (attributeVal == 0) {
        printf("Device %d doesn't support VIRTUAL ADDRESS MANAGEMENT.\n", cuDevice);
        exit(EXIT_WAIVED);
    }

    // The vector addition happens on cuDevice, so the allocations need to be
    // mapped there.
    vector<CUdevice> mappingDevices;
    mappingDevices.push_back(cuDevice);

    // Collect devices accessible by the mapping device (cuDevice) into the
    // backingDevices vector.
    vector<CUdevice>  backingDevices  = getBackingDevices(cuDevice);
    CUctxCreateParams ctxCreateParams = {};

    // Create context
    checkCudaErrors(cuCtxCreate(&cuContext, &ctxCreateParams, 0, cuDevice));

    // first search for the module path before we load the results
    string module_path;

    std::ostringstream fatbin;

    if (!findFatbinPath(FATBIN_FILE, module_path, argv, fatbin)) {
        exit(EXIT_FAILURE);
    }
    else {
        printf("> initCUDA loading module: <%s>\n", module_path.c_str());
    }

    if (!fatbin.str().size()) {
        printf("fatbin file empty. exiting..\n");
        exit(EXIT_FAILURE);
    }

    // Create module from binary file (FATBIN)
    checkCudaErrors(cuModuleLoadData(&cuModule, fatbin.str().c_str()));

    // Get function handle from module
    checkCudaErrors(cuModuleGetFunction(&vecAdd_kernel, cuModule, "VecAdd_kernel"));

    // Allocate input vectors h_A and h_B in host memory