csrc/adam/multi_tensor_adam.cu · 31f46feee2d491d58a13404e354440551de9d5bf · OpenDAS / deepspeed

Jeff Rasley authored Nov 12, 2020


Co-authored-by: Shaden Smith <Shaden.Smith@microsoft.com>
Co-authored-by: Reza Yazdani <reyazda@microsoft.com>

31f46fee

multi_tensor_adam.cu 6.37 KB