Skip to content

check torch.is_grad_enabled before calling customer flash atten ops#1397

Closed
XiaobingSuper wants to merge 1 commit intoDao-AILab:mainfrom XiaobingSuper:xiaobing/reduce_memory
+6-14

Commits