Fix torch build error.