summaryrefslogtreecommitdiff
path: root/numpy/core/tests
diff options
context:
space:
mode:
authorRaghuveer Devulapalli <raghuveer.devulapalli@intel.com>2019-07-08 21:30:30 -0700
committerRaghuveer Devulapalli <raghuveer.devulapalli@intel.com>2019-09-27 15:39:51 -0700
commit7a327d09344214ba5aac95f2428bffaa7fa260d2 (patch)
tree7019aeeea52c309fd8c37482ba9ac260fab7134b /numpy/core/tests
parent344b40f70d83f1fe364d6a8cbc44aceea20d6e26 (diff)
downloadnumpy-7a327d09344214ba5aac95f2428bffaa7fa260d2.tar.gz
ENH: use AVX for floor, rint, ceil and trunc
Diffstat (limited to 'numpy/core/tests')
-rw-r--r--numpy/core/tests/test_umath.py10
1 files changed, 7 insertions, 3 deletions
diff --git a/numpy/core/tests/test_umath.py b/numpy/core/tests/test_umath.py
index 86c724b5a..264eea00e 100644
--- a/numpy/core/tests/test_umath.py
+++ b/numpy/core/tests/test_umath.py
@@ -742,10 +742,14 @@ class TestSpecialFloats(object):
assert_raises(FloatingPointError, np.reciprocal, np.array(-0.0, dtype=dt))
# func : [maxulperror, low, high]
-avx_ufuncs = {'sqrt' :[1, 0., 100.],
+avx_ufuncs = {'sqrt' :[1, 0., 100.],
'absolute' :[0, -100., 100.],
- 'reciprocal' :[1, 1., 100.],
- 'square' :[1, -100., 100.]}
+ 'reciprocal' :[1, 1., 100.],
+ 'square' :[1, -100., 100.],
+ 'rint' :[0, -100., 100.],
+ 'floor' :[0, -100., 100.],
+ 'ceil' :[0, -100., 100.],
+ 'trunc' :[0, -100., 100.]}
class TestAVXUfuncs(object):
def test_avx_based_ufunc(self):