Are you outputting an rgb image from your code device?
RGB data is stored as fp16. So it is reduced in precision compared to heightfield data. This is a great tradeoff for operating on actual image data, but it is not quite enough precision to use for geometry purposes.
The recommended workaround for the above right now is to instead export two heightfields from your code device.