ColossalAI

Author	SHA1	Message	Date
Jiarui Fang	1aad903c15	[tensor] redistribute among different process groups (#1247 ) * make it faster * [tensor] rename convert_to_dist -> redistribute * [tensor] ShardSpec and ReplicaSpec * [tensor] redistribute among diff pgs * polish code	2 years ago
Jiarui Fang	20da6e48c8	[checkpoint] save sharded optimizer states (#1237 )	2 years ago
HELSON	f071b500b6	[polish] polish __repr__ for ColoTensor, DistSpec, ProcessGroup (#1235 )	2 years ago
Jiarui Fang	a98319f023	[tensor] torch function return colotensor (#1229 )	2 years ago
HELSON	280a81243d	[tensor] improve robustness of class 'ProcessGroup' (#1223 )	2 years ago
Jiarui Fang	15d988f954	[tensor] sharded global process group (#1219 )	2 years ago
Jiarui Fang	ae7d3f4927	[refactor] move process group from _DistSpec to ColoTensor. (#1203 )	2 years ago
Jiarui Fang	b5f25eb32a	[Tensor] add cpu group to ddp (#1200 )	2 years ago
Jiarui Fang	060b917daf	[refactor] remove gpc dependency in colotensor's _ops (#1189 )	2 years ago
Jiarui Fang	c463f8adf9	[tensor] remove gpc in tensor tests (#1186 )	2 years ago
Jiarui Fang	7487215b95	[ColoTensor] add independent process group (#1179 )	2 years ago