YAKL_error.h

Go to the documentation of this file.

 
#pragma once
// Included by YAKL.h
 
__YAKL_NAMESPACE_WRAPPER_BEGIN__
namespace yakl {
 
  YAKL_INLINE void yakl_throw(const char * msg) {
    // If we're on the host, then let's throw a real exception
    YAKL_EXECUTE_ON_HOST_ONLY(
      fence();
      std::cerr << "YAKL FATAL ERROR:\n";
      std::cerr << msg << std::endl;
      throw std::runtime_error(msg);
    )
    // Otherwise, we need to be more careful with printf and intentionally segfaulting to stop the program
    YAKL_EXECUTE_ON_DEVICE_ONLY(
      #ifdef YAKL_ARCH_SYCL
        // SYCL cannot printf like the other backends quite yet
        const CL_CONSTANT char format[] = "KERNEL CHECK FAILED:\n   %s\n";
        sycl::ext::oneapi::experimental::printf(format,msg);
      #elif defined(YAKL_ARCH_CUDA) || defined(YAKL_ARCH_HIP)
        printf("%s\n",msg);
      #endif
      // Intentionally cause a segfault to kill the run
      int *segfault = nullptr;
      *segfault = 10;
    )
  }
 
 
  inline void check_last_error() {
    #ifdef YAKL_DEBUG
      fence();
      #ifdef YAKL_ARCH_CUDA
        auto ierr = cudaGetLastError();
        if (ierr != cudaSuccess) { yakl_throw( cudaGetErrorString( ierr ) ); }
      #elif defined(YAKL_ARCH_HIP)
        auto ierr = hipGetLastError();
        if (ierr != hipSuccess) { yakl_throw( hipGetErrorString( ierr ) ); }
      #elif defined(YAKL_ARCH_SYCL)
      #endif
    #endif
  }
 
 
  inline bool yakl_mainproc() {
    // Only actually check if the user says MPI is available. Otherwise, always return true
    #ifdef HAVE_MPI
      int is_initialized;
      MPI_Initialized(&is_initialized);
      if (!is_initialized) {
        return true;
      } else {
        int myrank;
        MPI_Comm_rank(MPI_COMM_WORLD, &myrank);
        return myrank == 0;
      }
    #else
      return true;
    #endif
  }
 
}
__YAKL_NAMESPACE_WRAPPER_END__