KernelMatrix interface fix
Thrust fix