cu_util.h

Go to the documentation of this file.

#ifndef GHOST_CU_UTIL_H
#define GHOST_CU_UTIL_H

#include "config.h"
#include "types.h"
#include "error.h"
#ifdef GHOST_HAVE_CUDA
#include <cublas_v2.h>
#include <curand.h>
#include <cusparse_v2.h>
#endif

#ifdef GHOST_HAVE_CUDA
typedef cublasHandle_t ghost_cublas_handle_t;
typedef cusparseHandle_t ghost_cusparse_handle_t;
typedef struct cudaDeviceProp ghost_cu_deviceprop;
typedef curandGenerator_t ghost_cu_rand_generator;
#else
typedef int ghost_cublas_handle_t;
typedef int ghost_cusparse_handle_t;
typedef int ghost_cu_deviceprop;
typedef int ghost_cu_rand_generator;
#endif

typedef struct {
    int ndistinctdevice;
    int *ndevice;
    char **names;
} ghost_gpu_info;

#ifdef __cplusplus
extern "C" {
#endif

ghost_error ghost_cu_init(int dev);
ghost_error ghost_cu_malloc_managed(void **mem, size_t bytesize);
ghost_error ghost_cu_malloc(void **mem, size_t bytesize);
ghost_error ghost_cu_malloc_mapped(void **mem, const size_t size);
ghost_error ghost_cu_malloc_pinned(void **mem, const size_t size);
ghost_error ghost_cu_download(void *hostmem, void *devmem, size_t bytesize);
ghost_error ghost_cu_download2d(void *dest, size_t dpitch, const void *src, size_t spitch, size_t width, size_t height);
ghost_error ghost_cu_upload(void *devmem, void *hostmem, size_t bytesize);
ghost_error ghost_cu_upload2d(void *dest, size_t dpitch, const void *src, size_t spitch, size_t width, size_t height);
ghost_error ghost_cu_memcpy(void *dest, void *src, size_t bytesize);
ghost_error ghost_cu_memcpy2d(void *dest, size_t dpitch, const void *src, size_t spitch, size_t width, size_t height);
ghost_error ghost_cu_memset(void *s, int c, size_t n);
ghost_error ghost_cu_free(void *mem);
ghost_error ghost_cu_free_host(void *mem);
ghost_error ghost_cu_barrier();
ghost_error ghost_cu_ndevice(int *devcount);
ghost_error ghost_cu_version(int *ver);
ghost_error ghost_cu_gpu_info_create(ghost_gpu_info **gpu_info);
ghost_error ghost_cu_device(int *device);
ghost_error ghost_cu_cublas_handle(ghost_cublas_handle_t *handle);
ghost_error ghost_cu_cusparse_handle(ghost_cusparse_handle_t *handle);
ghost_error ghost_cu_deviceprop_get(ghost_cu_deviceprop *prop);

ghost_error ghost_cu_rand_generator_get(ghost_cu_rand_generator *gen);
ghost_error ghost_cu_finalize();
ghost_error ghost_cu_memtranspose(int torows, int tocols, void *to, int ldto, const void *from, int ldfrom, ghost_datatype dt);

ghost_error ghost_cu_reduce(void *out, void *data, ghost_datatype dt, ghost_lidx n);
ghost_error ghost_cu_reduce_multiple(void *out, void *data, ghost_datatype dt, ghost_lidx n, ghost_lidx ncols);

#ifdef __cplusplus
}
#endif

#endif