1#ifndef __CS_BASE_CUDA_H__
2#define __CS_BASE_CUDA_H__
48#define CS_CUDA_CHECK(a) { \
49 cudaError_t ret_code = a; \
50 if (cudaSuccess != ret_code) { \
51 bft_error(__FILE__, __LINE__, 0, "[CUDA error] %d: %s\n running: %s", \
52 ret_code, ::cudaGetErrorString(ret_code), #a); \
56#define CS_CUDA_CHECK_CALL(a, file_name, line_num) { \
57 cudaError_t ret_code = a; \
58 if (cudaSuccess != ret_code) { \
59 bft_error(file_name, line_num, 0, "[CUDA error] %d: %s\n running: %s", \
60 ret_code, ::cudaGetErrorString(ret_code), #a); \
68#define CS_CUDA_WARP_SIZE 32
122cs_cuda_mem_malloc_device(
size_t n,
123 const char *var_name,
124 const char *file_name,
148cs_cuda_mem_malloc_host(
size_t n,
149 const char *var_name,
150 const char *file_name,
174cs_cuda_mem_malloc_managed(
size_t n,
175 const char *var_name,
176 const char *file_name,
200cs_cuda_mem_free(
void *
p,
201 const char *var_name,
202 const char *file_name,
226cs_cuda_mem_free_host(
void *
p,
227 const char *var_name,
228 const char *file_name,
246cs_cuda_copy_h2d(
void *dst,
268cs_cuda_copy_h2d_async(
void *dst,
289cs_cuda_copy_d2h(
void *dst,
310cs_cuda_copy_d2h_async(
void *dst,
331cs_cuda_prefetch_h2d(
void *dst,
351cs_cuda_prefetch_d2h(
void *dst,
369cs_cuda_copy_d2d(
void *dst,
392cs_cuda_get_host_ptr(
const void *ptr);
415static inline unsigned int
417 unsigned int block_size)
419 return (n % block_size) ? n/block_size + 1 : n/block_size;
426#if defined(HAVE_CUDA)
437cs_base_cuda_device_info(
cs_log_t log_id);
448cs_base_cuda_version_info(
cs_log_t log_id);
459cs_base_cuda_compiler_info(
cs_log_t log_id);
473cs_base_cuda_select_default_device(
void);
484cs_base_cuda_get_device(
void);
static unsigned int cs_cuda_grid_size(cs_lnum_t n, unsigned int block_size)
Compute grid size for given array and block sizes.
Definition cs_base_cuda.h:416
int cs_glob_cuda_max_threads_per_block
int cs_glob_cuda_max_blocks
int cs_glob_cuda_device_id
int cs_glob_cuda_max_block_size
#define BEGIN_C_DECLS
Definition cs_defs.h:509
#define END_C_DECLS
Definition cs_defs.h:510
int cs_lnum_t
local mesh entity id
Definition cs_defs.h:313
@ p
Definition cs_field_pointer.h:67
cs_log_t
Definition cs_log.h:48