extensions/optimizer/fused_optimizer_cuda.py · 7cfed5f076f683293eb559c7e61feb6d3376b8d3 · OpenDAS / ColossalAI

Frank Lee authored Jan 25, 2024

* [feat] refactored extension module

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

7cfed5f0

fused_optimizer_cuda.py 1.08 KB