From 86d655526964bd66232e4ec7de3a564dbb51ebe1 Mon Sep 17 00:00:00 2001
From: cossio <j.cossio.diaz@gmail.com>
Date: Thu, 7 May 2020 09:54:05 +0200
Subject: [PATCH] cufunc

---
 src/layers/stateless.jl | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/layers/stateless.jl b/src/layers/stateless.jl
index 13e165f0..57c7398d 100644
--- a/src/layers/stateless.jl
+++ b/src/layers/stateless.jl
@@ -271,6 +271,10 @@ function xlogx(x)
   result = x * log(x)
   ifelse(x > zero(x), result, zero(result))
 end
+CuArrays.@cufunc function xlogx(x)
+  result = x * log(x)
+  ifelse(x > zero(x), result, zero(result))
+end
 
 """
     xlogy(x, y)
@@ -280,3 +284,7 @@ function xlogy(x, y)
   result = x * log(y)
   ifelse(x > zero(x), result, zero(result))
 end
+CuArrays.@cufunc function xlogy(x, y)
+  result = x * log(y)
+  ifelse(x > zero(x), result, zero(result))
+end