github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-29 04:40:36 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
595bedf7678d1d157e7a8e8230e266ddf21a26d3
ColossalAI/tests/test_tensor
History
Jiarui Fang 294a6060d0 [tensor] ZeRO use ColoTensor as the base class. (#828) 
* [refactor] moving InsertPostInitMethodToModuleSubClasses to utils.

* [tensor] ZeRO use ColoTensor as the base class.

* polish
2022-04-22 12:00:48 +08:00
..
test_op.py
[tensor] ZeRO use ColoTensor as the base class. (#828)
2022-04-22 12:00:48 +08:00