diff --git a/test/test_wrapper.py b/test/test_wrapper.py index 1f13b9cd9e12bd4497a33a14b956c4b86ed44c5b..61c6d45a50d715aa11c686b6acd98a63c4e35390 100644 --- a/test/test_wrapper.py +++ b/test/test_wrapper.py @@ -668,7 +668,10 @@ def test_enqueue_task(ctx_factory): assert la.norm(a[::-1] - b) == 0 -def test_platform_get_devices(platform): +def test_platform_get_devices(ctx_factory): + ctx = ctx_factory() + platform = ctx.devices[0].platform + if platform.name == "Apple": pytest.xfail("Apple doesn't understand all the values we pass " "for dev_type") @@ -676,9 +679,12 @@ def test_platform_get_devices(platform): dev_types = [cl.device_type.ACCELERATOR, cl.device_type.ALL, cl.device_type.CPU, cl.device_type.DEFAULT, cl.device_type.GPU] if (platform._get_cl_version() >= (1, 2) and - cl.get_cl_header_version() >= (1, 2)): + cl.get_cl_header_version() >= (1, 2) + and not platform.name.lower().startswith("nvidia")): dev_types.append(cl.device_type.CUSTOM) + for dev_type in dev_types: + print(dev_type) devs = platform.get_devices(dev_type) if dev_type in (cl.device_type.DEFAULT, cl.device_type.ALL,