fusion3: torch fix

Change-Id: I30e4daa3d2b33b78c50218cf9ff38c884a62000a