I have Mi3 and Mi4 running on the same rom, and I have that heat issue in Mi4. My Mi3 is cold, but it's a different story with Mi4. And one more thing, my keyboard became laggy after 5.5.8.
Edited: Laggy keyboard has nothing to do with SMIUI, it because of Xposed Installer. I switched to WSM...