1#ifndef __CS_BASE_CUDA_H__
2#define __CS_BASE_CUDA_H__
51#define CS_CUDA_CHECK(a) { \
52 cudaError_t _l_ret_code = a; \
53 if (cudaSuccess != _l_ret_code) { \
54 bft_error(__FILE__, __LINE__, 0, "[CUDA error] %d: %s\n running: %s", \
55 _l_ret_code, ::cudaGetErrorString(_l_ret_code), #a); \
59#define CS_CUDA_CHECK_CALL(a, file_name, line_num) { \
60 cudaError_t _l_ret_code = a; \
61 if (cudaSuccess != _l_ret_code) { \
62 bft_error(file_name, line_num, 0, "[CUDA error] %d: %s\n running: %s", \
63 _l_ret_code, ::cudaGetErrorString(_l_ret_code), #a); \
71#define CS_CUDA_WARP_SIZE 32
85extern int cs_glob_cuda_device_id;
89extern int cs_glob_cuda_max_threads_per_block;
90extern int cs_glob_cuda_max_block_size;
91extern int cs_glob_cuda_max_blocks;
92extern int cs_glob_cuda_n_mp;
97extern bool cs_glob_cuda_allow_graph;
122cs_cuda_copy_h2d(
void *dst,
144cs_cuda_copy_h2d_async(
void *dst,
165cs_cuda_copy_d2h(
void *dst,
186cs_cuda_copy_d2h_async(
void *dst,
205cs_cuda_copy_d2d(
void *dst,
228cs_cuda_get_host_ptr(
const void *ptr);
249static inline unsigned int
251 unsigned int block_size)
253 return (n % block_size) ? n/block_size + 1 : n/block_size;
275cs_sync_or_copy_h2d(
const T *val_h,
282 const T *_val_d = NULL;
288 size_t size = n_vals *
sizeof(T);
291 CS_CUDA_CHECK(cudaMalloc(&_buf_d, size));
292 cs_cuda_copy_h2d(_buf_d, val_h, size);
293 _val_d = (
const T *)_buf_d;
335cs_cuda_get_stream(
int stream_id);
348cs_cuda_get_stream_prefetch(
void);
363cs_base_cuda_device_info(
cs_log_t log_id);
374cs_base_cuda_version_info(
cs_log_t log_id);
385cs_base_cuda_compiler_info(
cs_log_t log_id);
399cs_base_cuda_select_default_device(
void);
410cs_base_cuda_get_device(
void);
#define BEGIN_C_DECLS
Definition: cs_defs.h:542
#define END_C_DECLS
Definition: cs_defs.h:543
int cs_lnum_t
local mesh entity id
Definition: cs_defs.h:335
cs_log_t
Definition: cs_log.h:48
static const void * cs_get_device_ptr_const(const void *ptr)
Return matching device pointer for a given constant pointer.
Definition: cs_mem.h:693
static cs_alloc_mode_t cs_check_device_ptr(const void *ptr)
Check if a pointer is associated with a device.
Definition: cs_mem.h:787
static void cs_sync_h2d(const void *ptr)
Synchronize data from host to device.
Definition: cs_mem.h:997
cs_alloc_mode_t
Definition: cs_mem.h:50
@ CS_ALLOC_HOST
Definition: cs_mem.h:52
@ CS_ALLOC_HOST_DEVICE_SHARED
Definition: cs_mem.h:57