Add Back Torch for DU