Shared gpu on cpu

Can we condition also on `device=gpu` here?