CudaAccelerator.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2025 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* CudaAccelerator.h                                           (C) 2000-2025 */
/*                                                                           */
/* Backend 'CUDA' pour les accélérateurs.                                    */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_CUDA_CUDAACCELERATOR_H
#define ARCANE_CUDA_CUDAACCELERATOR_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/Array.h"
#include "arcane/utils/TraceInfo.h"
#include "arcane/utils/NotSupportedException.h"
#include "arcane/utils/FatalErrorException.h"
 
#include <iostream>
 
#include <cuda_runtime.h>
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#ifdef ARCANE_COMPONENT_arcane_cuda
#define ARCANE_CUDA_EXPORT ARCANE_EXPORT
#else
#define ARCANE_CUDA_EXPORT ARCANE_IMPORT
#endif
 
namespace Arcane::Accelerator::Cuda
{
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
extern "C++" ARCANE_CUDA_EXPORT void
arcaneCheckCudaErrors(const TraceInfo& ti,cudaError_t e);
 
extern "C++" ARCANE_CUDA_EXPORT void
arcaneCheckCudaErrorsNoThrow(const TraceInfo& ti,cudaError_t e);

#define ARCANE_CHECK_CUDA(result) \
  Arcane::Accelerator::Cuda::arcaneCheckCudaErrors(A_FUNCINFO,result)

#define ARCANE_CHECK_CUDA_NOTHROW(result) \
  Arcane::Accelerator::Cuda::arcaneCheckCudaErrorsNoThrow(A_FUNCINFO,result)
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
extern "C++" ARCANE_CUDA_EXPORT Arccore::IMemoryAllocator*
getCudaMemoryAllocator();

extern "C++" ARCANE_CUDA_EXPORT Arccore::IMemoryAllocator*
getCudaDeviceMemoryAllocator();

extern "C++" ARCANE_CUDA_EXPORT Arccore::IMemoryAllocator*
getCudaUnifiedMemoryAllocator();

extern "C++" ARCANE_CUDA_EXPORT Arccore::IMemoryAllocator*
getCudaHostPinnedMemoryAllocator();
 
extern "C++" ARCANE_CUDA_EXPORT void
initializeCudaMemoryAllocators();
 
extern "C++" ARCANE_CUDA_EXPORT void
finalizeCudaMemoryAllocators(ITraceMng* tm);
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#if defined(CUDART_VERSION) && CUDART_VERSION >= 13000
#define ARCANE_USING_CUDA13_OR_GREATER
#endif
 
// A partir de CUDA 13, il y a un nouveau type cudaMemLocation
// pour les méthodes telles cudeMemAdvise ou cudaMemPrefetch
#if defined(ARCANE_USING_CUDA13_OR_GREATER)
inline cudaMemLocation
_getMemoryLocation(int device_id)
{
  cudaMemLocation mem_location;
  mem_location.type = cudaMemLocationTypeDevice;
  mem_location.id = device_id;
  if (device_id == cudaCpuDeviceId)
    mem_location.type = cudaMemLocationTypeHost;
  else {
    mem_location.type = cudaMemLocationTypeDevice;
    mem_location.id = device_id;
  }
  return mem_location;
}
#else
inline int
_getMemoryLocation(int device_id)
{
  return device_id;
}
#endif
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
} // End namespace Arcane::accelerator::cuda
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif