1 files changed, 27 insertions, 0 deletions
diff --git a/ot/gpu/cudamat/test/test_learn.py b/ot/gpu/cudamat/test/test_learn.py
new file mode 100644
index 0000000..954757a
--- /dev/null
+++ b/ot/gpu/cudamat/test/test_learn.py
@@ -0,0 +1,27 @@
+import pdb
+import numpy as np
+import nose
+import cudamat as cm
+
+def setup():
+    cm.cublas_init()
+
+def teardown():
+    cm.cublas_shutdown()
+
+def test_mult_by_sigmoid_deriv():
+    m = 256
+    n = 128
+    c_targets = np.array(np.random.randn(m, n)*10, dtype=np.float32, order='F')
+    c_acts = np.array(np.random.rand(m, n), dtype=np.float32, order='F')
+
+    g_targets = cm.CUDAMatrix(c_targets)
+    g_acts = cm.CUDAMatrix(c_acts)
+
+    c_targets = c_targets * c_acts * (1. - c_acts)
+    cm.learn.mult_by_sigmoid_deriv(g_targets, g_acts)
+
+    assert np.max(np.abs(c_acts - g_acts.asarray())) < 10**-2, "Error in cudamat.learn.mult_by_sigmoid_deriv exceeded threshold"
+
+if __name__ == '__main__':
+    nose.runmodule()