ColossalAI

Author	SHA1	Message	Date
HELSON	552183bb74	[polish] polish ColoTensor and its submodules (#2537 )	2023-02-03 11:44:10 +08:00
Jiarui Fang	36824a304c	[Doc] add more doc for ColoTensor. (#1458 )	2022-08-16 10:38:41 +08:00
Jiarui Fang	a1476ea882	[NFC] polish doc style for ColoTensor (#1457 )	2022-08-16 09:21:05 +08:00
Jiarui Fang	c92f84fcdb	[tensor] distributed checkpointing for parameters (#1240 )	2022-07-12 15:51:06 +08:00
HELSON	f071b500b6	[polish] polish __repr__ for ColoTensor, DistSpec, ProcessGroup (#1235 )	2022-07-08 13:25:57 +08:00
Jiarui Fang	ae7d3f4927	[refactor] move process group from _DistSpec to ColoTensor. (#1203 )	2022-07-06 16:15:16 +08:00
Jiarui Fang	060b917daf	[refactor] remove gpc dependency in colotensor's _ops (#1189 )	2022-07-04 18:54:37 +08:00
Jiarui Fang	0dd4e2bbfb	[Tensor] rename some APIs in TensorSpec and Polish view unittest (#1176 )	2022-06-27 15:56:11 +08:00
Jiarui Fang	07f9c781f9	[graph] improve the graph building. (#1157 )	2022-06-22 16:47:20 +08:00
Ziyue Jiang	6c5996a56e	[Tensor] add module check and bert test (#1031 ) * add Embedding * Add bert test * polish * add check module test * polish * polish * polish * polish	2022-05-26 18:15:42 +08:00
ver217	ad536e308e	[tensor] refactor colo-tensor (#992 ) * refactor colo-tensor and update linear op * polish code * polish code * update ops and unit tests * update unit tests * polish code * rename dist_spec module * polish code * polish code * remove unneeded import * fix pipelinable	2022-05-19 12:44:59 +08:00