2 Commits (8241c0c054b38a109ed3ce7be1052a1e600b8471)

Author SHA1 Message Date
Hongxin Liu 079bf3cb26
[misc] update pre-commit and run all files (#4752) 1 year ago
Hongxin Liu b5f9e37c70
[legacy] clean up legacy code (#4743) 1 year ago
digger-yu b9a8dff7e5
[doc] Fix typo under colossalai and doc(#3618) 2 years ago
HELSON 552183bb74
[polish] polish ColoTensor and its submodules (#2537) 2 years ago
Jiarui Fang 36824a304c
[Doc] add more doc for ColoTensor. (#1458) 2 years ago
Jiarui Fang a1476ea882
[NFC] polish doc style for ColoTensor (#1457) 2 years ago
Jiarui Fang c92f84fcdb
[tensor] distributed checkpointing for parameters (#1240) 2 years ago
HELSON f071b500b6
[polish] polish __repr__ for ColoTensor, DistSpec, ProcessGroup (#1235) 2 years ago
Jiarui Fang ae7d3f4927
[refactor] move process group from _DistSpec to ColoTensor. (#1203) 2 years ago
Jiarui Fang 060b917daf
[refactor] remove gpc dependency in colotensor's _ops (#1189) 2 years ago
Jiarui Fang 0dd4e2bbfb
[Tensor] rename some APIs in TensorSpec and Polish view unittest (#1176) 2 years ago
Jiarui Fang 07f9c781f9
[graph] improve the graph building. (#1157) 2 years ago
Ziyue Jiang 6c5996a56e
[Tensor] add module check and bert test (#1031) 3 years ago
ver217 ad536e308e
[tensor] refactor colo-tensor (#992) 3 years ago
ver217 c2fdc6a011
[tensor] derive compute pattern from dist spec (#971) 3 years ago
ver217 67c33f57eb
[tensor] design DistSpec and DistSpecManager for ColoTensor (#934) 3 years ago