ColossalAI/tests/test_auto_parallel
Boyuan Yao a2b43e393d
[autoparallel] Patch meta information of `torch.nn.Embedding` (#2760)
* [autoparallel] embedding metainfo

* [autoparallel] fix function name in test_activation_metainfo

* [autoparallel] undo changes in activation metainfo and related tests
2023-02-17 10:39:48 +08:00
..
test_pass [autoparallel] refactor runtime pass (#2644) 2023-02-15 10:36:19 +08:00
test_tensor_shard [autoparallel] Patch meta information of `torch.nn.Embedding` (#2760) 2023-02-17 10:39:48 +08:00
__init__.py [autoparallel] handled illegal sharding strategy (#1728) 2022-10-19 12:53:06 +08:00