static int hasCublasError(const char * msg) { cublasStatus err = cublasGetError(); if(err != CUBLAS_STATUS_SUCCESS) error("cublas error : %s : %s\n", msg, cublasGetErrorString(err)); return 0; }
TEST_F(CudaAdapterTest, CublasGetErrorString) { EXPECT_EQ("CUBLAS_STATUS_SUCCESS", cublasGetErrorString(CUBLAS_STATUS_SUCCESS)); EXPECT_EQ("CUBLAS_STATUS_NOT_INITIALIZED", cublasGetErrorString(CUBLAS_STATUS_NOT_INITIALIZED)); EXPECT_EQ("CUBLAS_STATUS_ALLOC_FAILED", cublasGetErrorString(CUBLAS_STATUS_ALLOC_FAILED)); EXPECT_EQ("CUBLAS_STATUS_INVALID_VALUE", cublasGetErrorString(CUBLAS_STATUS_INVALID_VALUE)); EXPECT_EQ("CUBLAS_STATUS_ARCH_MISMATCH", cublasGetErrorString(CUBLAS_STATUS_ARCH_MISMATCH)); EXPECT_EQ("CUBLAS_STATUS_MAPPING_ERROR", cublasGetErrorString(CUBLAS_STATUS_MAPPING_ERROR)); EXPECT_EQ("CUBLAS_STATUS_EXECUTION_FAILED", cublasGetErrorString(CUBLAS_STATUS_EXECUTION_FAILED)); EXPECT_EQ("CUBLAS_STATUS_INTERNAL_ERROR", cublasGetErrorString(CUBLAS_STATUS_INTERNAL_ERROR)); }
static void checkCublasError(const char * msg) { cublasStatus err = cublasGetError(); if(err != CUBLAS_STATUS_SUCCESS) error("cublas error : %s : %s\n", msg, cublasGetErrorString(err)); }