fix build error torch