Skip to content
Snippets Groups Projects
Commit 3f59645f authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Fix usable_local_mem_size() query.

parent c878c8de
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment