diff options
author | Raghuveer Devulapalli <raghuveer.devulapalli@intel.com> | 2019-07-08 21:30:30 -0700 |
---|---|---|
committer | Raghuveer Devulapalli <raghuveer.devulapalli@intel.com> | 2019-09-27 15:39:51 -0700 |
commit | 7a327d09344214ba5aac95f2428bffaa7fa260d2 (patch) | |
tree | 7019aeeea52c309fd8c37482ba9ac260fab7134b /numpy/core/tests | |
parent | 344b40f70d83f1fe364d6a8cbc44aceea20d6e26 (diff) | |
download | numpy-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.py | 10 |
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): |