cublas_check.hpp

Go to the documentation of this file.

 
#pragma once
 
#include <cublas_v2.h>
 
#include <stdexcept>
#include <string>
 
#define PULSATRIX_CUBLAS_CHECK(call)                                                                   \
    do {                                                                                            \
        cublasStatus_t pulsatrix_cublas_check_result = (call);                                            \
        if (pulsatrix_cublas_check_result != CUBLAS_STATUS_SUCCESS) {                                      \
            throw std::runtime_error(std::string("cuBLAS error: ") +                                   \
                                      cublasGetStatusString(pulsatrix_cublas_check_result) + " at " __FILE__ \
                                      + ":" + std::to_string(__LINE__));                                  \
        }                                                                                                   \
    } while (0)