diff options
Diffstat (limited to 'ot/gpu/cudamat/test/test_learn.py')
-rw-r--r-- | ot/gpu/cudamat/test/test_learn.py | 27 |
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() |