cuda.h

// ---------------------------------------------------------------------
//
// Copyright (C) 2018 by the deal.II authors
//
// This file is part of the deal.II library.
//
// The deal.II library 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 deal.II distribution.
//
// ---------------------------------------------------------------------

#ifndef dealii_cuda_h
#define dealii_cuda_h

#include <deal.II/base/config.h>

#ifdef DEAL_II_WITH_CUDA

#include <deal.II/base/exceptions.h>

#include <cusolverDn.h>
#include <cusolverSp.h>
#include <cusparse.h>
#include <vector>

DEAL_II_NAMESPACE_OPEN
namespace Utilities
{
  namespace CUDA
  {
    struct Handle
    {
      Handle();

      Handle(Handle const &) = delete;

      ~Handle();

      cusolverDnHandle_t cusolver_dn_handle;

      cusolverSpHandle_t cusolver_sp_handle;

      cusparseHandle_t cusparse_handle;
    };

    template <typename T>
    inline void malloc(T *&pointer, const unsigned int n_elements)
    {
      cudaError_t cuda_error_code = cudaMalloc(&pointer, n_elements * sizeof(T));
      AssertCuda(cuda_error_code);
    }

    template <typename T>
    inline void free(T *&pointer)
    {
      cudaError_t cuda_error_code = cudaFree(pointer);
      AssertCuda(cuda_error_code);
      pointer = nullptr;
    }

    template <typename T>
    inline void copy_to_host(const T *pointer_dev,
                             std::vector<T> &vector_host)
    {
      cudaError_t cuda_error_code =
        cudaMemcpy(vector_host.data(), pointer_dev,
                   vector_host.size() * sizeof(T), cudaMemcpyDeviceToHost);
      AssertCuda(cuda_error_code);
    }

    template <typename T>
    inline void copy_to_dev(const std::vector<T> &vector_host,
                            T *pointer_dev)
    {
      cudaError_t cuda_error_code =
        cudaMemcpy(pointer_dev, vector_host.data(),
                   vector_host.size() * sizeof(T), cudaMemcpyHostToDevice);
      AssertCuda(cuda_error_code);
    }
  }
}

DEAL_II_NAMESPACE_CLOSE

#endif

#endif