From c0a17bc98afc83dd2cf2448433cce10f56aae820 Mon Sep 17 00:00:00 2001 From: Reazul Hoque Date: Thu, 17 Dec 2020 03:09:55 -0600 Subject: [PATCH] Skip some tests on Gen12 (#120) Co-authored-by: reazul.hoque Co-authored-by: Sergey Pokhodenko --- numba_dppy/tests/skip_tests.py | 11 +++++++++++ numba_dppy/tests/test_numpy_math_functions.py | 6 +++++- .../tests/test_numpy_trigonomteric_functions.py | 2 ++ 3 files changed, 18 insertions(+), 1 deletion(-) create mode 100644 numba_dppy/tests/skip_tests.py diff --git a/numba_dppy/tests/skip_tests.py b/numba_dppy/tests/skip_tests.py new file mode 100644 index 0000000000..fa18d36181 --- /dev/null +++ b/numba_dppy/tests/skip_tests.py @@ -0,0 +1,11 @@ +import dpctl + +def is_gen12(device_type): + with dpctl.device_context(device_type): + q = dpctl.get_current_queue() + device = q.get_sycl_device() + name = device.get_device_name() + if "Gen12" in name: + return True + + return False diff --git a/numba_dppy/tests/test_numpy_math_functions.py b/numba_dppy/tests/test_numpy_math_functions.py index cf5174ac96..ef5dc235b8 100644 --- a/numba_dppy/tests/test_numpy_math_functions.py +++ b/numba_dppy/tests/test_numpy_math_functions.py @@ -3,7 +3,7 @@ from numba import njit import dpctl import unittest - +from . import skip_tests @unittest.skipUnless(dpctl.has_gpu_queues(), 'test only on GPU system') class TestNumpy_math_functions(unittest.TestCase): @@ -179,6 +179,7 @@ def f(a): self.assertTrue(np.all(c == -input_arr)) + @unittest.skipIf(skip_tests.is_gen12("opencl:gpu"), "Gen12 not supported") def test_sign(self): @njit def f(a): @@ -221,6 +222,7 @@ def f(a): max_abs_err = c.sum() - d.sum() self.assertTrue(max_abs_err < 1e-5) + @unittest.skipIf(skip_tests.is_gen12("opencl:gpu"), "Gen12 not supported") def test_log(self): @njit def f(a): @@ -236,6 +238,7 @@ def f(a): max_abs_err = c.sum() - d.sum() self.assertTrue(max_abs_err < 1e-5) + @unittest.skipIf(skip_tests.is_gen12("opencl:gpu"), "Gen12 not supported") def test_log10(self): @njit def f(a): @@ -251,6 +254,7 @@ def f(a): max_abs_err = c.sum() - d.sum() self.assertTrue(max_abs_err < 1e-5) + @unittest.skipIf(skip_tests.is_gen12("opencl:gpu"), "Gen12 not supported") def test_expm1(self): @njit def f(a): diff --git a/numba_dppy/tests/test_numpy_trigonomteric_functions.py b/numba_dppy/tests/test_numpy_trigonomteric_functions.py index 7c0a95d323..812f3d060c 100644 --- a/numba_dppy/tests/test_numpy_trigonomteric_functions.py +++ b/numba_dppy/tests/test_numpy_trigonomteric_functions.py @@ -3,6 +3,7 @@ from numba import njit import dpctl import unittest +from . import skip_tests @unittest.skipUnless(dpctl.has_gpu_queues(), 'test only on GPU system') @@ -155,6 +156,7 @@ def f(a): max_abs_err = c.sum() - d.sum() self.assertTrue(max_abs_err < 1e-5) + @unittest.skipIf(skip_tests.is_gen12("opencl:gpu"), "Gen12 not supported") def test_arccosh(self): @njit def f(a):