Como faço para passar um array bidimensional para um kernel no pycuda?

Eu encontrei uma respostaAqui, mas não está claro se devo remodelar o array. Preciso remodelar a matriz 2d em 1d antes de passá-la ao kernel pycuda?