CudaHelpers.h

Go to the documentation of this file.

/*
* SPDX-FileCopyrightText: Copyright (c) 2020-2021 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
* SPDX-License-Identifier: MIT
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
* DEALINGS IN THE SOFTWARE.
*/
 
 
#ifndef NVNEURAL_CUDAHELPERS_H
#define NVNEURAL_CUDAHELPERS_H
 
#include <nvneural/CudaTypes.h>
#include <nvneural/CoreHelpers.h>
#include <cassert>
#include <cuda_runtime_api.h>
 
namespace nvneural {
 
const size_t STANDARD_CUDA_BLOCK_SIZE = 32;
 
inline NeuralResult CheckCudartErrors_Impl(cudaError_t status, const char file[], int line)
{
    if (status != cudaSuccess)
    {
        DefaultLogger()->logError(0, "CUDA error: %s (error %d)", cudaGetErrorString(status), status);
        DefaultLogger()->logError(1, "Location: %s, line %d", file, line);
        if ((status == cudaErrorInsufficientDriver) ||
            (status == cudaErrorInitializationError))
        {
            DefaultLogger()->logError(0, "Driver is not supported");
        }
 
        return NeuralResult::Failure;
    }
 
    return NeuralResult::Success;
}
 
#define CheckCudartErrors(status_) \
    do                                                                                            \
    {                                                                                             \
        const auto localResult = nvneural::CheckCudartErrors_Impl((status_), __FILE__, __LINE__); \
        if (failed(localResult))                                                                  \
        {                                                                                         \
            return localResult;                                                                   \
        }                                                                                         \
    } while (false)
 
inline NeuralResult CheckCudaDriverErrors_Impl(CUresult status, const char file[], int line)
{
    if (status != CUDA_SUCCESS)
    {
        DefaultLogger()->logError(0, "(CU)DA error: %d", status);
        DefaultLogger()->logError(1, "Location: %s, line %d", file, line);
 
        return NeuralResult::Failure;
    }
 
    return NeuralResult::Success;
}
 
#define CheckCudaDriverErrors(status_)                                                                \
    do                                                                                                \
    {                                                                                                 \
        const auto localResult = nvneural::CheckCudaDriverErrors_Impl((status_), __FILE__, __LINE__); \
        if (failed(localResult))                                                                      \
        {                                                                                             \
            return localResult;                                                                       \
        }                                                                                             \
    } while (false)
 
} // namespace nvneural
 
#endif // NVNEURAL_CUDAHELPERS_H