fix gpu_to_cpu support