diff --git a/pycuda/tools.py b/pycuda/tools.py index 9c213940d9ebdb6b253d5d9188442fe238cefa88..c243b911bf9a4331d358364c5bfac84b647aebce 100644 --- a/pycuda/tools.py +++ b/pycuda/tools.py @@ -40,6 +40,7 @@ import numpy as np bitlog2 = _drv.bitlog2 DeviceMemoryPool = _drv.DeviceMemoryPool PageLockedMemoryPool = _drv.PageLockedMemoryPool +PageLockedAllocator = _drv.PageLockedAllocator from pycuda.compyte.dtypes import ( register_dtype, get_or_register_dtype, _fill_dtype_registry,