summaryrefslogtreecommitdiff
path: root/tests/matrix_test.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/matrix_test.py')
-rw-r--r--tests/matrix_test.py14
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'