From 7ac468a1ea03beddf245b5c2597d91baafe2ec02 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 12 Jun 2013 00:41:10 -0400
Subject: [PATCH] Fix a few broken CL device type queries
---
loopy/auto_test.py | 2 +-
test/test_linalg.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/loopy/auto_test.py b/loopy/auto_test.py
index c28583695..b9b65f50d 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 c9605e5ee..374b43e6f 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
--
GitLab