6 Commits

Author	SHA1	Message	Date
Jiarui Fang	1f99205827	[Gemini] remove static tracer (#2083)	2022-12-06 12:53:58 +08:00
Jiarui Fang	2e9cbfca12	[Gemini] add unitests to check gemini correctness (#2015)	2022-11-24 16:51:45 +08:00
Jiarui Fang	3ce4463fe6	[utils] remove lazy_memory_allocate from ColoInitContext (#1844)	2022-11-09 11:50:33 +08:00
HELSON	7a065dc9f6	[hotfix] fix megatron_init in test_gpt2.py (#1357)	2022-07-25 10:28:19 +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
HELSON	bf5066fba7	[refactor] refactor ColoTensor's unit tests (#1340)	2022-07-19 15:46:24 +08:00