体验新版 GitCode,发现更多精彩内容 >>
Created by: mapingshuo
New features
OPs
Add swap mem op to copy mem btw CPU and GPU.
工作中发现有些Op的输出比较占显存,SwapMem可以把这些Var在前向时存在内存中,反向需要时取出到显存中即可。