[atlaas] cleanup includes and move length_sq