DeviceExceptions.cu.h

Go to the documentation of this file.

// ---------------------------------------------------------------------
//
// Copyright (c) 2017-2025 The Regents of the University of Michigan and DFT-FE
// authors.
//
// This file is part of the DFT-FE code.
//
// The DFT-FE code is free software; you can use it, redistribute
// it, and/or modify it under the terms of the GNU Lesser General
// Public License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
// The full text of the license can be found in the file LICENSE at
// the top level of the DFT-FE distribution.
//
// ---------------------------------------------------------------------
//
#ifndef dftefeDeviceExceptions_cuh
#define dftefeDeviceExceptions_cuh
 
 
#define DEVICE_API_CHECK(cmd)                                \
  do                                                         \
    {                                                        \
      cudaError_t e = cmd;                                   \
      if (e != cudaSuccess)                                  \
        {                                                    \
          printf("Failed: Cuda error in %s at %s:%d '%s'\n", \
                 __func__,                                   \
                 __FILE__,                                   \
                 __LINE__,                                   \
                 cudaGetErrorString(e));                     \
          exit(EXIT_FAILURE);                                \
        }                                                    \
    }                                                        \
  while (0)
 
#define DEVICEBLAS_API_CHECK(expr)                                                   \
  {                                                                                  \
    cublasStatus_t __cublas_error = expr;                                            \
    if ((__cublas_error) != CUBLAS_STATUS_SUCCESS)                                   \
      {                                                                              \
        printf(                                                                      \
          "cuBLAS error on or before line number %d in file: %s. Error code: %d.\n", \
          __LINE__,                                                                  \
          __FILE__,                                                                  \
          __cublas_error);                                                           \
      }                                                                              \
  }
 
 
#endif // dftefeDeviceExceptions_cuh