From cff59ba1353b79ec3b69059ce1b4f191540616fd Mon Sep 17 00:00:00 2001
From: Joseph Redmon <pjreddie@gmail.com>
Date: Wed, 16 Mar 2016 11:30:48 +0000
Subject: [PATCH] go updates
---
src/convolutional_kernels.cu | 11 ++---------
1 files changed, 2 insertions(+), 9 deletions(-)
diff --git a/src/convolutional_kernels.cu b/src/convolutional_kernels.cu
index 85b92df..3dc125f 100644
--- a/src/convolutional_kernels.cu
+++ b/src/convolutional_kernels.cu
@@ -65,9 +65,9 @@
}
}
-void binarize_filters_gpu(float *filters, int n, int size, float *mean)
+void binarize_filters_gpu(float *filters, int n, int size, float *binary)
{
- binarize_filters_kernel<<<cuda_gridsize(n), BLOCK>>>(filters, n, size, mean);
+ binarize_filters_kernel<<<cuda_gridsize(n), BLOCK>>>(filters, n, size, binary);
check_error(cudaPeekAtLastError());
}
@@ -161,13 +161,6 @@
check_error(cudaPeekAtLastError());
}
-void swap_binary(convolutional_layer *l)
-{
- float *swap = l->filters_gpu;
- l->filters_gpu = l->binary_filters_gpu;
- l->binary_filters_gpu = swap;
-}
-
void forward_convolutional_layer_gpu(convolutional_layer l, network_state state)
{
int i;
--
Gitblit v1.10.0