Only need your valid I-94 and passport to come back from Canada if you go Canada from US for short time and you don't go to other country(China) from Canada.