ColossalAI/tests/test_tensor
ver217 26b7aac0be
[zero] reorganize zero/gemini folder structure (#3424)
* [zero] refactor low-level zero folder structure

* [zero] fix legacy zero import path

* [zero] fix legacy zero import path

* [zero] remove useless import

* [zero] refactor gemini folder structure

* [zero] refactor gemini folder structure

* [zero] refactor legacy zero import path

* [zero] refactor gemini folder structure

* [zero] refactor gemini folder structure

* [zero] refactor gemini folder structure

* [zero] refactor legacy zero import path

* [zero] fix test import path

* [zero] fix test

* [zero] fix circular import

* [zero] update import
2023-04-04 13:48:16 +08:00
..
common_utils [zero] add unit testings for hybrid parallelism (#2486) 2023-01-18 10:36:10 +08:00
core [colotensor] add Tensor.view op and its unit test (#1343) 2022-07-21 10:53:15 +08:00
model [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
test_dtensor [DTensor] refactor dtensor with new components (#3089) 2023-03-14 16:25:47 +08:00
test_colo_checkpoint_tools.py [colotensor] use cpu memory to store state_dict (#1367) 2022-07-26 14:13:38 +08:00
test_comm_spec_apply.py [autoparallel] shard param and buffer as expected (#1753) 2022-10-21 15:45:13 +08:00
test_context.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00
test_mix_gather.py [autoparallel] mix gather (#1977) 2022-11-23 21:49:17 +08:00
test_parameter.py [refactor] refactor ColoTensor's unit tests (#1340) 2022-07-19 15:46:24 +08:00
test_shape_consistency.py [autoparallel] update CommSpec (#1667) 2022-09-29 11:20:59 +08:00
test_shape_consistency_apply.py [autoparallel] shard param and buffer as expected (#1753) 2022-10-21 15:45:13 +08:00
test_sharded_linear.py [ColoTensor] ColoInitContext initialize parameters in shard mode. (#1937) 2022-11-14 16:05:09 +08:00
test_sharding_spec.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00
test_tp_with_zero.py [zero] reorganize zero/gemini folder structure (#3424) 2023-04-04 13:48:16 +08:00