mmr/tkDNN
Archived
1
0
Fork 0
Code Issues 46 Pull Requests 12 Releases 4 Activity

test ready

Browse Source
This commit is contained in:
Francesco Gatti
2017-06-29 12:26:51 +00:00
parent ba215c35c0
commit 0a0acb9b1d
3 changed files with 17 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/kernels/activation_elu.cu
+5 -2
View File
@@ -9,7 +9,7 @@
__global__
void activation_elu(value_type *input, value_type *output, int size) {
int i = threadIdx.x*(blockIdx.x +1);
int i = blockDim.x*blockIdx.x + threadIdx.x;
if(i<size) {
value_type k0, k1;
@@ -30,6 +30,9 @@ void activation_elu(value_type *input, value_type *output, int size) {
*/
void activationELUForward(value_type* srcData, value_type* dstData, int size)
{
activation_elu<<<(size+255)/256, 256>>>(srcData, dstData, size);
int blocks = (size+255)/256;
int threads = 256;
activation_elu<<<blocks, threads>>>(srcData, dstData, size);
checkCuda( cudaDeviceSynchronize() );
}