trats2: enable ums support on Trats2