From 3f59645fcc0eb77603ded43326f491ea1124e3a4 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sat, 16 Jul 2011 13:30:11 -0400
Subject: [PATCH] Fix usable_local_mem_size() query.
---
pyopencl/characterize.py | 12 +++++++++---
1 file changed, 9 insertions(+), 3 deletions(-)
diff --git a/pyopencl/characterize.py b/pyopencl/characterize.py
index a3b4c608..27a8438a 100644
--- a/pyopencl/characterize.py
+++ b/pyopencl/characterize.py
@@ -49,9 +49,15 @@ def usable_local_mem_size(dev, nargs=None):
"""
usable_local_mem_size = dev.local_mem_size
- if ("nvidia" in dev.platform.name.lower()
- and (dev.compute_capability_major_nv,
- dev.compute_capability_minor_nv) < (2, 0)):
+ try:
+ nv_compute_cap = (dev.compute_capability_major_nv,
+ dev.compute_capability_minor_nv)
+ except AttributeError:
+ nv_compute_cap = None
+
+
+ if (nv_compute_cap is not None
+ and nv_compute_cap < (2,0)):
# pre-Fermi use local mem for parameter passing
if nargs is None:
# assume maximum
--
GitLab