ensure the right device is chosen in 2D