Support yolov4

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-04-28 20:35:30 +02:00
parent a874fad2bd
commit db0f8a4d99
15 changed files with 1979 additions and 9 deletions
@@ -0,0 +1,16 @@
+#include "kernels.h"
+#include <math.h>
+
+__global__ void scal_add_kernel(dnnType* dstData, int size, float alpha, float beta, int inc)
+{
+    int i = (blockIdx.x + blockIdx.y*gridDim.x) * blockDim.x + threadIdx.x;
+    if (i < size) dstData[i*inc] = dstData[i*inc] * alpha + beta;
+}
+
+void scalAdd(dnnType* dstData, int size, float alpha, float beta, int inc, cudaStream_t stream)
+{
+    int blocks = (size+255)/256;
+    int threads = 256;
+    
+    scal_add_kernel<<<blocks, threads, 0, stream>>>(dstData, size, alpha, beta, inc);
+}