Merge pull request #52 from untom/mempool_py3

Fix PooledDeviceAllocation for Python3
inducer · Aug 25, 2014 · 6a8bcdd · 6a8bcdd
2 parents ac3bfc9 + 78b01c1
commit 6a8bcdd
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)
       ;