diff --git a/pyopencl/characterize.py b/pyopencl/characterize.py index a3b4c60863d0b5915067ad1e63bf42c282048e11..27a8438a0f095cbde81edcc601bab012ea850dd6 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