device.hpp

Go to the documentation of this file.

// Copyright (c) 2023--present, The Simons Foundation
// This file is part of TRIQS/nda and is licensed under the Apache License, Version 2.0.
// SPDX-License-Identifier: Apache-2.0
// See LICENSE in the root of this distribution for details.

 
#pragma once
 
#ifdef NDA_HAVE_CUDA
#include "./exceptions.hpp"
 
#include <cuda_runtime.h>
#include <cublas_v2.h>
 
#include <complex>
#include <exception>
#include <string>
#endif // NDA_HAVE_CUDA
 
namespace nda {


  template <bool flag = false>
  void compile_error_no_gpu() {
    static_assert(flag, "Using device functionality without gpu support! Configure project with -DCudaSupport=ON.");
  }
 
#ifdef NDA_HAVE_CUDA

  static constexpr bool have_device = true;

  static constexpr bool have_cuda = true;

  inline void device_error_check(cudaError_t success, std::string message = "") {
    if (success != cudaSuccess) {
      NDA_RUNTIME_ERROR << "Cuda runtime error: " << std::to_string(success) << "\n"
                        << " message: " << message << "\n"
                        << " cudaGetErrorName: " << std::string(cudaGetErrorName(success)) << "\n"
                        << " cudaGetErrorString: " << std::string(cudaGetErrorString(success)) << "\n";
    }
  }

  inline cublasOperation_t get_cublas_op(char op) {
    switch (op) {
      case 'N': return CUBLAS_OP_N;
      case 'T': return CUBLAS_OP_T;
      case 'C': return CUBLAS_OP_C;
      default: std::terminate(); return {};
    }
  }

  inline auto cucplx(std::complex<double> c) { return cuDoubleComplex{c.real(), c.imag()}; }

  inline auto *cucplx(std::complex<double> *c) { return reinterpret_cast<cuDoubleComplex *>(c); } // NOLINT

  inline auto *cucplx(std::complex<double> const *c) { return reinterpret_cast<const cuDoubleComplex *>(c); } // NOLINT

  inline auto **cucplx(std::complex<double> **c) { return reinterpret_cast<cuDoubleComplex **>(c); } // NOLINT

  inline auto **cucplx(std::complex<double> const **c) { return reinterpret_cast<const cuDoubleComplex **>(c); } // NOLINT
 
#else

#define device_error_check(ARG1, ARG2) compile_error_no_gpu()

  static constexpr bool have_device = false;

  static constexpr bool have_cuda = false;
 
#endif // NDA_HAVE_CUDA

 
} // namespace nda