diff --git a/pyopencl/mempool.py b/pyopencl/mempool.py index bd804799c95d921bd1aea60babf82f18c791200d..8ca837d0a8a87009879797d9c9f9025b2451826c 100644 --- a/pyopencl/mempool.py +++ b/pyopencl/mempool.py @@ -27,6 +27,7 @@ THE SOFTWARE. import numpy as np import pyopencl as cl +from pyopencl.tools import bitlog2 # {{{ allocators @@ -120,7 +121,7 @@ class MemoryPool(object): @classmethod def bin_number(cls, size): - l = max(size.bit_length(), 1) - 1 + l = bitlog2(size) mantissa_bits = cls.mantissa_bits if l >= mantissa_bits: