modules/gpu/src/cuda/safe_call.hpp

   1 /*M///////////////////////////////////////////////////////////////////////////////////////
   2 //
   3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
   4 //
   5 //  By downloading, copying, installing or using the software you agree to this license.
   6 //  If you do not agree to this license, do not download, install,
   7 //  copy or use the software.
   8 //
   9 //
  10 //                           License Agreement
  11 //                For Open Source Computer Vision Library
  12 //
  13 // Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
  14 // Copyright (C) 2009, Willow Garage Inc., all rights reserved.
  15 // Third party copyrights are property of their respective owners.
  16 //
  17 // Redistribution and use in source and binary forms, with or without modification,
  18 // are permitted provided that the following conditions are met:
  19 //
  20 //   * Redistribution's of source code must retain the above copyright notice,
  21 //     this list of conditions and the following disclaimer.
  22 //
  23 //   * Redistribution's in binary form must reproduce the above copyright notice,
  24 //     this list of conditions and the following disclaimer in the documentation
  25 //     and/or other materials provided with the distribution.
  26 //
  27 //   * The name of the copyright holders may not be used to endorse or promote products
  28 //     derived from this software without specific prior written permission.
  29 //
  30 // This software is provided by the copyright holders and contributors "as is" and
  31 // any express or implied warranties, including, but not limited to, the implied
  32 // warranties of merchantability and fitness for a particular purpose are disclaimed.
  33 // In no event shall the Intel Corporation or contributors be liable for any direct,
  34 // indirect, incidental, special, exemplary, or consequential damages
  35 // (including, but not limited to, procurement of substitute goods or services;
  36 // loss of use, data, or profits; or business interruption) however caused
  37 // and on any theory of liability, whether in contract, strict liability,
  38 // or tort (including negligence or otherwise) arising in any way out of
  39 // the use of this software, even if advised of the possibility of such damage.
  40 //
  41 //M*/
  42
  43 #ifndef __OPENCV_CUDA_SAFE_CALL_HPP__
  44 #define __OPENCV_CUDA_SAFE_CALL_HPP__
  45
  46 #include "cvconfig.h"
  47
  48 #include <cuda_runtime_api.h>
  49
  50 #ifdef HAVE_CUFFT
  51 #   include <cufft.h>
  52 #endif
  53
  54 #ifdef HAVE_CUBLAS
  55 #   include <cublas.h>
  56 #endif
  57
  58 #include "NCV.hpp"
  59
  60 namespace cv { namespace gpu {
  61
  62 void nppError(int err, const char *file, const int line, const char *func = "");
  63
  64 void ncvError(int err, const char *file, const int line, const char *func = "");
  65
  66 #ifdef HAVE_CUFFT
  67     void cufftError(int err, const char *file, const int line, const char *func = "");
  68 #endif
  69
  70 #ifdef HAVE_CUBLAS
  71     void cublasError(int err, const char *file, const int line, const char *func = "");
  72 #endif
  73
  74 }}
  75
  76 // nppSafeCall
  77
  78 static inline void ___nppSafeCall(int err, const char *file, const int line, const char *func = "")
  79 {
  80     if (err < 0)
  81         cv::gpu::nppError(err, file, line, func);
  82 }
  83
  84 #if defined(__GNUC__)
  85     #define nppSafeCall(expr)  ___nppSafeCall(expr, __FILE__, __LINE__, __func__)
  86 #else
  87     #define nppSafeCall(expr)  ___nppSafeCall(expr, __FILE__, __LINE__)
  88 #endif
  89
  90 // ncvSafeCall
  91
  92 static inline void ___ncvSafeCall(int err, const char *file, const int line, const char *func = "")
  93 {
  94     if (NCV_SUCCESS != err)
  95         cv::gpu::ncvError(err, file, line, func);
  96 }
  97
  98 #if defined(__GNUC__)
  99     #define ncvSafeCall(expr)  ___ncvSafeCall(expr, __FILE__, __LINE__, __func__)
 100 #else
 101     #define ncvSafeCall(expr)  ___ncvSafeCall(expr, __FILE__, __LINE__)
 102 #endif
 103
 104 // cufftSafeCall
 105
 106 #ifdef HAVE_CUFFT
 107     static inline void ___cufftSafeCall(cufftResult_t err, const char *file, const int line, const char *func = "")
 108     {
 109         if (CUFFT_SUCCESS != err)
 110             cv::gpu::cufftError(err, file, line, func);
 111     }
 112
 113     #if defined(__GNUC__)
 114         #define cufftSafeCall(expr)  ___cufftSafeCall(expr, __FILE__, __LINE__, __func__)
 115     #else
 116         #define cufftSafeCall(expr)  ___cufftSafeCall(expr, __FILE__, __LINE__)
 117     #endif
 118 #endif
 119
 120 // cublasSafeCall
 121
 122 #ifdef HAVE_CUBLAS
 123     static inline void ___cublasSafeCall(cublasStatus_t err, const char *file, const int line, const char *func = "")
 124     {
 125         if (CUBLAS_STATUS_SUCCESS != err)
 126             cv::gpu::cublasError(err, file, line, func);
 127     }
 128
 129     #if defined(__GNUC__)
 130         #define cublasSafeCall(expr)  ___cublasSafeCall(expr, __FILE__, __LINE__, __func__)
 131     #else
 132         #define cublasSafeCall(expr)  ___cublasSafeCall(expr, __FILE__, __LINE__)
 133     #endif
 134 #endif
 135
 136 #endif /* __OPENCV_CUDA_SAFE_CALL_HPP__ */