Some fixes to get CUDA assignment kernels running on the device