Using the PyCUDA module