diff --git a/src/kernels/activations.cu b/src/kernels/activations.cu index 0e8be1e..0ceffd7 100644 --- a/src/kernels/activations.cu +++ b/src/kernels/activations.cu @@ -24,13 +24,3 @@ relu_kernel(const float* __restrict__ src, float* __restrict__ dst, int len) { dst[i] = src[i] < 0.0 ? 0.0 : src[i]; } } - -__global__ void -linear_kernel(const float* __restrict__ src, float* __restrict__ dst, int len) { - int stride = gridDim.x * blockDim.x; - int tid = blockDim.x * blockIdx.x + threadIdx.x; - - for (int i = tid; i < len; i += stride) { - dst[i] = src[i]; - } -} diff --git a/src/layers/dense.cu b/src/layers/dense.cu index 47cb48c..138d456 100644 --- a/src/layers/dense.cu +++ b/src/layers/dense.cu @@ -75,9 +75,6 @@ void Layers::Dense::forward(const float* d_input, float* d_output) { break; default: - linear_kernel<<<1, outputSize>>>( - d_output, d_output, outputSize - ); break; }