|
|
|
@ -36,19 +36,19 @@ extern int gpu_index; |
|
|
|
|
#ifdef __cplusplus |
|
|
|
|
extern "C" { |
|
|
|
|
#endif // __cplusplus
|
|
|
|
|
void check_error(cudaError_t status); |
|
|
|
|
cublasHandle_t blas_handle(); |
|
|
|
|
float *cuda_make_array(float *x, size_t n); |
|
|
|
|
int *cuda_make_int_array(size_t n); |
|
|
|
|
void cuda_push_array(float *x_gpu, float *x, size_t n); |
|
|
|
|
void cuda_pull_array(float *x_gpu, float *x, size_t n); |
|
|
|
|
YOLODLL_API void cuda_set_device(int n); |
|
|
|
|
int cuda_get_device(); |
|
|
|
|
void cuda_free(float *x_gpu); |
|
|
|
|
void cuda_random(float *x_gpu, size_t n); |
|
|
|
|
float cuda_compare(float *x_gpu, float *x, size_t n, char *s); |
|
|
|
|
dim3 cuda_gridsize(size_t n); |
|
|
|
|
cudaStream_t get_cuda_stream(); |
|
|
|
|
void check_error(cudaError_t status); |
|
|
|
|
cublasHandle_t blas_handle(); |
|
|
|
|
float *cuda_make_array(float *x, size_t n); |
|
|
|
|
int *cuda_make_int_array(size_t n); |
|
|
|
|
void cuda_push_array(float *x_gpu, float *x, size_t n); |
|
|
|
|
YOLODLL_API void cuda_pull_array(float *x_gpu, float *x, size_t n); |
|
|
|
|
YOLODLL_API void cuda_set_device(int n); |
|
|
|
|
int cuda_get_device(); |
|
|
|
|
void cuda_free(float *x_gpu); |
|
|
|
|
void cuda_random(float *x_gpu, size_t n); |
|
|
|
|
float cuda_compare(float *x_gpu, float *x, size_t n, char *s); |
|
|
|
|
dim3 cuda_gridsize(size_t n); |
|
|
|
|
cudaStream_t get_cuda_stream(); |
|
|
|
|
#ifdef __cplusplus |
|
|
|
|
} |
|
|
|
|
#endif // __cplusplus
|
|
|
|
|