Draft: [surface] mixed precision improvements

Here I store some mixed precision improvements, however, we will first solve the problem of handling edge corner cases, before coming back to this.