diff --git a/loopy/auto_test.py b/loopy/auto_test.py index c2858369504ca2a097d933d9b6cea54393ecadcb..b9b65f50dc9c8d296bfec2adc1b10b130a9d0e62 100644 --- a/loopy/auto_test.py +++ b/loopy/auto_test.py @@ -300,7 +300,7 @@ def _enumerate_cl_devices_for_ref_test(): continue for dev in pf.get_devices(): - if dev.type == cl.device_type.CPU: + if dev.type & cl.device_type.CPU: cpu_devs.append(dev) else: noncpu_devs.append(dev) diff --git a/test/test_linalg.py b/test/test_linalg.py index c9605e5ee355f2be42e472e7c26d83269e09c1ac..374b43e6f2f1c75887800df541bdf5bdb88b21ab 100644 --- a/test/test_linalg.py +++ b/test/test_linalg.py @@ -46,7 +46,7 @@ DEBUG_PREAMBLE = r""" def get_suitable_size(ctx): dev, = ctx.devices - if dev.type == cl.device_type.CPU: + if dev.type & cl.device_type.CPU: return 160 else: return 1600