diff --git a/python/test/test_tensor_types.py b/python/test/test_tensor_types.py index 91866f885..fb11083fd 100644 --- a/python/test/test_tensor_types.py +++ b/python/test/test_tensor_types.py @@ -4,6 +4,7 @@ import pytest import kp import numpy as np +VK_ICD_FILENAMES = os.environ.get("VK_ICD_FILENAMES", "") def test_type_float(): @@ -82,7 +83,9 @@ def test_type_float_double_incorrect(): assert np.all(tensor_out.data() != arr_in_a * arr_in_b) -@pytest.mark.skipif("swiftshader" in os.environ.get("VK_ICD_FILENAMES", ""), +@pytest.mark.skipif("broadcom" in VK_ICD_FILENAMES, + reason="Broadcom doesn't support double") +@pytest.mark.skipif("swiftshader" in VK_ICD_FILENAMES, reason="Swiftshader doesn't support double") def test_type_double():