5 Commits

Author	SHA1	Message	Date
HELSON	552183bb74	[polish] polish ColoTensor and its submodules (#2537)	2023-02-03 11:44:10 +08:00
Jiarui Fang	a1476ea882	[NFC] polish doc style for ColoTensor (#1457)	2022-08-16 09:21:05 +08:00
HELSON	7a8702c06d	[colotensor] add Tensor.view op and its unit test (#1343) ... [colotensor] add megatron initialization for gpt2	2022-07-21 10:53:15 +08:00
Jiarui Fang	4b9bba8116	[ColoTensor] rename APIs and add output_replicate to ComputeSpec (#1168)	2022-06-24 13:08:54 +08:00
Jiarui Fang	f4ef224358	[Tensor] remove ParallelAction, use ComputeSpec instread (#1166)	2022-06-23 17:34:59 +08:00