summaryrefslogtreecommitdiff
path: root/ot/gpu/cudamat/test/test_learn.py
diff options
context:
space:
mode:
Diffstat (limited to 'ot/gpu/cudamat/test/test_learn.py')
-rw-r--r--ot/gpu/cudamat/test/test_learn.py27
1 files changed, 0 insertions, 27 deletions
diff --git a/ot/gpu/cudamat/test/test_learn.py b/ot/gpu/cudamat/test/test_learn.py
deleted file mode 100644
index 954757a..0000000
--- a/ot/gpu/cudamat/test/test_learn.py
+++ /dev/null
@@ -1,27 +0,0 @@
-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()