Fix pycuda compatiblity.