upsample ok

This commit is contained in:
Francesco Gatti
2018-12-20 17:08:31 +01:00
parent 2ab47b5874
commit c8f2e1b448
6 changed files with 72 additions and 6 deletions
+21
View File
@@ -0,0 +1,21 @@
#include "kernels.h"
__global__
void fill_kernel(dnnType *data, int size, dnnType val) {
int i = blockDim.x*blockIdx.x + threadIdx.x;
if(i<size) {
data[i] = val;
}
}
void fill(dnnType* data, int size, dnnType val, cudaStream_t stream)
{
int blocks = (size+255)/256;
int threads = 256;
fill_kernel<<<blocks, threads, 0, stream>>>(data, size, val);
}