From efbe95428988d479d3c7e14494858d96d5c2062f Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 21 Jul 2011 14:43:31 -0500
Subject: [PATCH] Make a way to obtain the NV compute capability.
---
pyopencl/characterize.py | 17 +++++++++++------
1 file changed, 11 insertions(+), 6 deletions(-)
diff --git a/pyopencl/characterize.py b/pyopencl/characterize.py
index 27a8438a..02afd223 100644
--- a/pyopencl/characterize.py
+++ b/pyopencl/characterize.py
@@ -43,18 +43,23 @@ def reasonable_work_group_size_multiple(dev, ctx=None):
+def get_nv_compute_capability(dev):
+ try:
+ return (dev.compute_capability_major_nv,
+ dev.compute_capability_minor_nv)
+ except:
+ return None
+
+
+
+
def usable_local_mem_size(dev, nargs=None):
"""Return an estimate of the usable local memory size.
:arg nargs: Number of 32-bit arguments passed.
"""
usable_local_mem_size = dev.local_mem_size
- try:
- nv_compute_cap = (dev.compute_capability_major_nv,
- dev.compute_capability_minor_nv)
- except AttributeError:
- nv_compute_cap = None
-
+ nv_compute_cap = get_nv_compute_capability(dev)
if (nv_compute_cap is not None
and nv_compute_cap < (2,0)):
--
GitLab