removed sync
This commit is contained in:
@@ -34,5 +34,4 @@ void activationELUForward(dnnType* srcData, dnnType* dstData, int size)
|
|||||||
int threads = 256;
|
int threads = 256;
|
||||||
|
|
||||||
activation_elu<<<blocks, threads>>>(srcData, dstData, size);
|
activation_elu<<<blocks, threads>>>(srcData, dstData, size);
|
||||||
checkCuda( cudaDeviceSynchronize() );
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -23,7 +23,6 @@ void activationLEAKYForward(dnnType* srcData, dnnType* dstData, int size)
|
|||||||
int threads = 256;
|
int threads = 256;
|
||||||
|
|
||||||
activation_leaky<<<blocks, threads>>>(srcData, dstData, size);
|
activation_leaky<<<blocks, threads>>>(srcData, dstData, size);
|
||||||
checkCuda( cudaDeviceSynchronize() );
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -20,7 +20,6 @@ void activationLOGISTICForward(dnnType* srcData, dnnType* dstData, int size)
|
|||||||
int threads = 256;
|
int threads = 256;
|
||||||
|
|
||||||
activation_logistic<<<blocks, threads>>>(srcData, dstData, size);
|
activation_logistic<<<blocks, threads>>>(srcData, dstData, size);
|
||||||
checkCuda( cudaDeviceSynchronize() );
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -44,7 +44,6 @@ void reorgForward(dnnType* srcData, dnnType* dstData,
|
|||||||
int threads = 256;
|
int threads = 256;
|
||||||
|
|
||||||
reorg_kernel<<<blocks, threads>>>(size, srcData, w, h, c, n, stride, false, dstData);
|
reorg_kernel<<<blocks, threads>>>(size, srcData, w, h, c, n, stride, false, dstData);
|
||||||
checkCuda( cudaDeviceSynchronize() );
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -39,5 +39,4 @@ void softmaxForward(float *input, int n, int batch, int batch_offset,
|
|||||||
int threads = 256;
|
int threads = 256;
|
||||||
|
|
||||||
softmax_kernel<<<blocks, threads>>>(input, n, batch, batch_offset, groups, group_offset, stride, temp, output);
|
softmax_kernel<<<blocks, threads>>>(input, n, batch, batch_offset, groups, group_offset, stride, temp, output);
|
||||||
checkCuda( cudaDeviceSynchronize() );
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user