There is no need to duplicate the code the enables TMU. Also update the
comment to better explain why we do this in the first place.

No functional changes.
Signed-off-by: Mika Westerberg <mika.westerberg@linux.intel.com>