From 168af40fe9a3cc81c6ee16b3e81f154780c36bdb Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期四, 03 六月 2021 15:03:27 +0800
Subject: [PATCH] up new v4
---
lib/detecter_tools/darknet/dark_cuda.h | 206 ++++++++++++++++++++++++++-------------------------
1 files changed, 106 insertions(+), 100 deletions(-)
diff --git a/lib/detecter_tools/darknet/dark_cuda.h b/lib/detecter_tools/darknet/dark_cuda.h
index d0581db..9251e87 100644
--- a/lib/detecter_tools/darknet/dark_cuda.h
+++ b/lib/detecter_tools/darknet/dark_cuda.h
@@ -1,100 +1,106 @@
-#ifndef DARKCUDA_H
-#define DARKCUDA_H
-#include "darknet.h"
-
-#ifdef __cplusplus
-extern "C" {
-#endif
-
-
-extern int cuda_debug_sync;
-extern int gpu_index;
-#ifdef __cplusplus
-}
-#endif // __cplusplus
-
-#ifdef GPU
-
-#define BLOCK 512
-#define FULL_MASK 0xffffffff
-#define WARP_SIZE 32
-#define BLOCK_TRANSPOSE32 256
-
-#include <cuda.h>
-#include <cuda_runtime.h>
-#include <curand.h>
-#include <cublas_v2.h>
-#include <cuda_runtime_api.h>
-//#include <driver_types.h>
-
-#ifdef CUDNN
-#include <cudnn.h>
-#endif // CUDNN
-
-#ifndef __DATE__
-#define __DATE__
-#endif
-
-#ifndef __TIME__
-#define __TIME__
-#endif
-
-#ifndef __FUNCTION__
-#define __FUNCTION__
-#endif
-
-#ifndef __LINE__
-#define __LINE__ 0
-#endif
-
-#ifndef __FILE__
-#define __FILE__
-#endif
-
-#ifdef __cplusplus
-extern "C" {
-#endif // __cplusplus
- void check_error(cudaError_t status);
- void check_error_extended(cudaError_t status, const char *file, int line, const char *date_time);
-#define CHECK_CUDA(X) check_error_extended(X, __FILE__ " : " __FUNCTION__, __LINE__, __DATE__ " - " __TIME__ );
-
- cublasHandle_t blas_handle();
- void free_pinned_memory();
- void pre_allocate_pinned_memory(size_t size);
- float *cuda_make_array_pinned_preallocated(float *x, size_t n);
- float *cuda_make_array_pinned(float *x, size_t n);
- float *cuda_make_array(float *x, size_t n);
- void **cuda_make_array_pointers(void **x, size_t n);
- int *cuda_make_int_array(size_t n);
- int *cuda_make_int_array_new_api(int *x, size_t n);
- void cuda_push_array(float *x_gpu, float *x, size_t n);
- //LIB_API void cuda_pull_array(float *x_gpu, float *x, size_t n);
- //LIB_API void cuda_set_device(int n);
- int cuda_get_device();
- void cuda_free_host(float *x_cpu);
- 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();
- cudaStream_t get_cuda_memcpy_stream();
- int get_number_of_blocks(int array_size, int block_size);
- int get_gpu_compute_capability(int i, char *device_name);
- void show_cuda_cudnn_info();
-
-#ifdef CUDNN
-cudnnHandle_t cudnn_handle();
-enum {cudnn_fastest, cudnn_smallest, cudnn_specify};
-
-void cudnn_check_error_extended(cudnnStatus_t status, const char *file, int line, const char *date_time);
-#define CHECK_CUDNN(X) cudnn_check_error_extended(X, __FILE__ " : " __FUNCTION__, __LINE__, __DATE__ " - " __TIME__ );
-#endif
-
-#ifdef __cplusplus
-}
-#endif // __cplusplus
-
-#else // GPU
-//LIB_API void cuda_set_device(int n);
-#endif // GPU
-#endif // DARKCUDA_H
+#ifndef DARKCUDA_H
+#define DARKCUDA_H
+#include "darknet.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+
+extern int cuda_debug_sync;
+extern int gpu_index;
+#ifdef __cplusplus
+}
+#endif // __cplusplus
+
+#ifdef GPU
+
+#define BLOCK 512
+#define FULL_MASK 0xffffffff
+#define WARP_SIZE 32
+#define BLOCK_TRANSPOSE32 256
+
+#include <cuda.h>
+#include <cuda_runtime.h>
+#include <curand.h>
+#include <cublas_v2.h>
+#include <cuda_runtime_api.h>
+//#include <driver_types.h>
+
+#ifdef CUDNN
+#include <cudnn.h>
+#endif // CUDNN
+
+#ifndef __DATE__
+#define __DATE__
+#endif
+
+#ifndef __TIME__
+#define __TIME__
+#endif
+
+#ifndef __FUNCTION__
+#define __FUNCTION__
+#endif
+
+#ifndef __LINE__
+#define __LINE__ 0
+#endif
+
+#ifndef __FILE__
+#define __FILE__
+#endif
+
+#ifdef __cplusplus
+extern "C" {
+#endif // __cplusplus
+ void check_error(cudaError_t status);
+ void check_error_extended(cudaError_t status, const char *file, int line, const char *date_time);
+ void cublas_check_error_extended(cublasStatus_t status, const char *file, int line, const char *date_time);
+#define CHECK_CUDA(X) check_error_extended(X, __FILE__ " : " __FUNCTION__, __LINE__, __DATE__ " - " __TIME__ );
+#define CHECK_CUBLAS(X) cublas_check_error_extended(X, __FILE__ " : " __FUNCTION__, __LINE__, __DATE__ " - " __TIME__ );
+
+ cublasHandle_t blas_handle();
+ void free_pinned_memory();
+ void pre_allocate_pinned_memory(size_t size);
+ float *cuda_make_array_pinned_preallocated(float *x, size_t n);
+ float *cuda_make_array_pinned(float *x, size_t n);
+ float *cuda_make_array(float *x, size_t n);
+ void **cuda_make_array_pointers(void **x, size_t n);
+ int *cuda_make_int_array(size_t n);
+ int *cuda_make_int_array_new_api(int *x, size_t n);
+ void cuda_push_array(float *x_gpu, float *x, size_t n);
+ //LIB_API void cuda_pull_array(float *x_gpu, float *x, size_t n);
+ //LIB_API void cuda_set_device(int n);
+ int cuda_get_device();
+ void cuda_free_host(float *x_cpu);
+ 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();
+ //cudaStream_t get_cuda_memcpy_stream();
+ int get_number_of_blocks(int array_size, int block_size);
+ int get_gpu_compute_capability(int i, char *device_name);
+ void show_cuda_cudnn_info();
+
+ cudaStream_t switch_stream(int i);
+ void wait_stream(int i);
+ void reset_wait_stream_events();
+
+#ifdef CUDNN
+cudnnHandle_t cudnn_handle();
+enum {cudnn_fastest, cudnn_smallest, cudnn_specify};
+
+void cudnn_check_error_extended(cudnnStatus_t status, const char *file, int line, const char *date_time);
+#define CHECK_CUDNN(X) cudnn_check_error_extended(X, __FILE__ " : " __FUNCTION__, __LINE__, __DATE__ " - " __TIME__ );
+#endif
+
+#ifdef __cplusplus
+}
+#endif // __cplusplus
+
+#else // GPU
+//LIB_API void cuda_set_device(int n);
+#endif // GPU
+#endif // DARKCUDA_H
--
Gitblit v1.8.0