diff options
Diffstat (limited to 'tests/matrix_test.py')
-rw-r--r-- | tests/matrix_test.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/tests/matrix_test.py b/tests/matrix_test.py index 412acf9..bc4115e 100644 --- a/tests/matrix_test.py +++ b/tests/matrix_test.py @@ -34,6 +34,7 @@ matrixsubequalsfloat = mod.get_function('matrixsubequalsfloat') matrixmulequalsfloat = mod.get_function('matrixmulequalsfloat') matrixdivequalsfloat = mod.get_function('matrixdivequalsfloat') outer = mod.get_function('outer') +minusmatrix = mod.get_function('minusmatrix') size = {'block': (1,1,1), 'grid': (1,1)} @@ -308,3 +309,16 @@ for i in range(1): print dest else: print 'success' + + print 'testing minus matrix...', + + dest = np.copy(a) + + minusmatrix(cuda.In(a), cuda.Out(dest), **size) + + if not np.allclose(-a, dest): + print 'fail' + print -a + print dest + else: + print 'success' |