Add some __restrict__ to the kernels to speed up compute capability 3.5 devices.