From dbdd31ee211fe8b1ac7e93ceadf7b34b8d304f34 Mon Sep 17 00:00:00 2001
From: Roland Singer <roland.singer@desertbit.com>
Date: Wed, 22 Aug 2018 11:56:41 +0000
Subject: [PATCH] updated README to include information about learning rate adjustment for multiple GPUs
---
src/cuda.h | 26 +++++++++++++-------------
1 files changed, 13 insertions(+), 13 deletions(-)
diff --git a/src/cuda.h b/src/cuda.h
index e167d36..50bb82b 100644
--- a/src/cuda.h
+++ b/src/cuda.h
@@ -36,19 +36,19 @@
#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
--
Gitblit v1.10.0