mirror of https://github.com/hpcaitech/ColossalAI
1f894e033f
* add placement policy * add gemini mgr * update mem stats collector * update zero * update zero optim * fix bugs * zero optim monitor os * polish unit test * polish unit test * add assert |
||
---|---|---|
.. | ||
_utils | ||
test_addmm_tp.py | ||
test_chunk.py | ||
test_context.py | ||
test_dist_spec_mgr.py | ||
test_embedding_tp.py | ||
test_gpt.py | ||
test_graph.py | ||
test_hybrid_device.py | ||
test_linear_tp.py | ||
test_model.py | ||
test_module_spec.py | ||
test_op.py | ||
test_parameter.py | ||
test_tensor.py | ||
test_zero.py | ||
test_zero_optim.py |