search

bytedance / lightseq

LightSeq: A High Performance Library for Sequence Processing and Generation
Other
3.22k stars 329 forks source link

fix compatibility with torch > 1.10 #359

Closed Taka152 closed 2 years ago

Taka152 commented 2 years ago

torch has removed <THC/THCGeneral.h> file, it will raise error when using ls_adam. Change to ATen's API to build with the newer torch.