add skip_when_disabled_cl_khr_fp16

fixstars · Feb 3, 2020 · 69c3a83 · 69c3a83
1 parent 1af1dea
commit 69c3a83
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/clpy/testing/__init__.py b/clpy/testing/__init__.py
@@ -13,6 +13,7 @@
 from clpy.testing.array import assert_array_max_ulp  # NOQA
 from clpy.testing.attr import gpu  # NOQA
 from clpy.testing.attr import multi_gpu  # NOQA
+from clpy.testing.attr import skip_when_disabled_cl_khr_fp16  # NOQA
 from clpy.testing.attr import slow  # NOQA
 from clpy.testing.bufio import readbuf  # NOQA
 from clpy.testing.bufio import writebuf  # NOQA

diff --git a/clpy/testing/attr.py b/clpy/testing/attr.py
@@ -1,3 +1,4 @@
+import clpy
 import os
 import unittest
 
@@ -65,3 +66,10 @@ def gpu(f):
 
     check_available()
     return multi_gpu(1)(pytest.mark.gpu(f))
+
+
+def skip_when_disabled_cl_khr_fp16(f):
+    check_available()
+    return unittest.skipUnless(
+        clpy.backend.opencl.env.supports_cl_khr_fp16() ==
+        clpy.backend.opencl.api.TRUE, "")(f)