Доработать код Python для запуска на PyCUDA

Есть код на Python, необходимо дописать его до состояния возможности запускать на GPU через PyCUDA