ColossalAI/tests/test_utils
HELSON e5ea3fdeef
[gemini] add GeminiMemoryManger (#832)
* refactor StatefulTensor, tensor utilities

* add unitest for GeminiMemoryManager
2022-04-24 13:08:48 +08:00
..
test_checkpoint
test_activation_checkpointing.py
test_commons.py [gemini] add GeminiMemoryManger (#832) 2022-04-24 13:08:48 +08:00
test_memory.py
test_pipelinable.py [pipelinable]use pipelinable context to initialize non-pipeline model (#816) 2022-04-24 13:03:12 +08:00
test_zero_gradient_clippling.py [test] refactored with the new rerun decorator (#763) 2022-04-15 00:33:04 +08:00