Fix PooledDeviceAllocation for Python3

inducer · Aug 25, 2014 · 78b01c1 · 78b01c1
1 parent ac3bfc9
commit 78b01c1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/wrapper/mempool.cpp b/src/wrapper/mempool.cpp
@@ -286,6 +286,7 @@ void pycuda_expose_tools()
       .DEF_SIMPLE_METHOD(free)
       .def("__int__", &cl::ptr)
       .def("__long__", pooled_device_allocation_to_long)
+      .def("__index__", pooled_device_allocation_to_long)
       .def("__len__", &cl::size)
       ;