1177 Commits (51b89d2202cebf7b81aeae220aa2699d2e012543)

 

All Branches   

Search

Author	SHA1	Message	Date
YuliangLiu0306	942c8cd1fb	[fx] refactor tracer to trace complete graph (#1342) ... * [CLI] add CLI launcher * Revert "[CLI] add CLI launcher" This reverts commit df7e6506d4. * [fx] refactor tracer to trace complete graph * add comments and solve conflicts.	2 years ago
Frank Lee	2cc1175c76	[fx] tested the complete workflow for auto-parallel (#1336) ... * [fx] tested the complete workflow for auto-parallel * polish code * polish code * polish code	2 years ago
YuliangLiu0306	4631fef8a0	[fx]refactor tracer (#1335)	2 years ago
HELSON	bf5066fba7	[refactor] refactor ColoTensor's unit tests (#1340)	2 years ago
HELSON	f92c100ddd	[checkpoint] use gather_tensor in checkpoint and update its unit test (#1339)	2 years ago
Frank Lee	f3ce7b8336	[fx] recovered skipped pipeline tests (#1338)	2 years ago
ver217	0c51ff2c13	[hotfix] ZeroDDP use new process group (#1333) ... * process group supports getting ranks in group * chunk mgr receives a process group * update unit test * fix unit tests	2 years ago
Frank Lee	11d1436a67	[workflow] update docker build workflow to use proxy (#1334)	2 years ago
Frank Lee	75abc75c15	[fx] fixed compatiblity issue with torch 1.10 (#1331)	2 years ago
Frank Lee	069d6fdc84	[workflow] update 8-gpu test to use torch 1.11 (#1332)	2 years ago
fastalgo	7857fd7616	Update README.md	2 years ago
Frank Lee	169954f87e	[test] removed outdated unit test for meta context (#1329)	2 years ago
ver217	7a05367101	[hotfix] shared model returns cpu state_dict (#1328)	2 years ago
Frank Lee	b2475d8c5c	[fx] fixed unit tests for torch 1.12 (#1327)	2 years ago
HELSON	d49708ae43	[hotfix] fix ddp for unit test test_gpt2 (#1326)	2 years ago
Frank Lee	250be4d31e	[utils] integrated colotensor with lazy init context (#1324) ... * [utils] integrated colotensor with lazy init context * polish code * polish code * polish code	2 years ago
Frank Lee	659a740738	[workflow] roll back to use torch 1.11 for unit testing (#1325)	2 years ago
Frank Lee	4d5dbf48a6	[workflow] fixed trigger condition for 8-gpu unit test (#1323)	2 years ago
YuliangLiu0306	e8acf55e8b	[fx] add balanced policy v2 (#1251) ... * [CLI] add CLI launcher * Revert "[CLI] add CLI launcher" This reverts commit df7e6506d4. * [fx] add balanced policy v2 * add unittest	2 years ago
XYE	ca2d3f284f	[fx] Add unit test and fix bugs for transform_mlp_pass (#1299) ... * add test and fix bugs * add functions back * add comments	2 years ago
HELSON	1b41686461	[hotfix] fix unit test test_module_spec (#1321)	2 years ago
Jiarui Fang	9e4c6449b0	[checkpoint] add ColoOptimizer checkpointing (#1316)	2 years ago
Frank Lee	7c2634f4b3	[workflow] updated release bdist workflow (#1318) ... * [workflow] updated release bdist workflow * polish workflow * polish workflow	2 years ago
github-actions[bot]	869cf3d3b8	Automated submodule synchronization (#1319) ... Co-authored-by: github-actions <github-actions@github.com>	2 years ago
Frank Lee	efdc240f1f	[workflow] disable SHM for compatibility CI on rtx3080 (#1315)	2 years ago
ver217	7c70bfbefa	[hotfix] fix PipelineSharedModuleGradientHandler (#1314)	2 years ago
Jiarui Fang	85f933b58b	[Optimizer] Remove useless ColoOptimizer (#1312)	2 years ago
Frank Lee	c9c37dcc4d	[workflow] updated pytorch compatibility test (#1311)	2 years ago
Jiarui Fang	9f10524313	[Optimizer] polish the init method of ColoOptimizer (#1310)	2 years ago
HELSON	36086927e1	[hotfix] fix ColoTensor GPT2 unitest (#1309)	2 years ago
Jiarui Fang	3ef3791a3b	[checkpoint] add test for bert and hotfix save bugs (#1297)	2 years ago
Jiarui Fang	bd71e2a88b	[hotfix] add missing file (#1308)	2 years ago
Frank Lee	4f4d8c3656	[fx] added apex normalization to patched modules (#1300) ... * [fx] added apex normalization to patched modules * remove unused imports	2 years ago
Jiarui Fang	4165eabb1e	[hotfix] remove potiential circle import (#1307) ... * make it faster * [hotfix] remove circle import	2 years ago
github-actions[bot]	6f2f9eb214	Automated submodule synchronization (#1305) ... Co-authored-by: github-actions <github-actions@github.com>	2 years ago
YuliangLiu0306	93a75433df	[hotfix] skip some unittest due to CI environment. (#1301)	2 years ago
lucasliunju	339520c6e0	[NFC] polish build_colossalai_wheel.py code style (#1306)	2 years ago
HELSON	260a55804a	[hotfix] fix shape error in backward when using ColoTensor (#1298)	2 years ago
runluo	f83c4d6597	[NFC] polish colossalai/nn/layer/wrapper/pipeline_wrapper.py code style (#1303)	2 years ago
binmakeswell	7696cead8d	Recover kernal files	2 years ago
XYE	e83b2ce853	[NFC] polish colossalai/nn/layer/vanilla/layers.py code style (#1295)	2 years ago
Liping233	1000a41fd5	[NFC] polish colossalai/nn/layer/vanilla/__init__.py code style (#1293)	2 years ago
Maruyama_Aya	87f679aeae	[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/kernels.h code style (#1291)	2 years ago
Wangbo Zhao(黑色枷锁)	552667825b	[NFC] polish colossalai/nn/layer/parallel_1d/layers.py code style (#1290)	2 years ago
doubleHU	d6f5ef8860	[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu code style (#1286)	2 years ago
Ziheng Qin	6d6c01e94d	[NFC] polish colossalai/__init__.py code style (#1285)	2 years ago
Jiatong Han	38e3ccd1e9	[NFC] polish colossalai/nn/layer/parallel_sequence/layers.py code style (#1280) ... Co-authored-by: JThh <jiatong.han@u.nus.edu>	2 years ago
Boyuan Yao	b414eaa5db	[NFC] polish colossalai/nn/optimizer/lamb.py code style (#1275)	2 years ago
yuxuan-lou	5f6ab35d25	Hotfix/format (#1274) ... * [NFC] Polish colossalai/kernel/cuda_native/csrc/multi_tensor_lamb.cu code style. (#937) * [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h code style * [NFC] polish colossalai/kernel/cuda_native/csrc/scaled_masked_softmax.cpp code style Co-authored-by: BoxiangW <45734921+BoxiangW@users.noreply.github.com>	2 years ago
Super Daniel	52d145a342	[NFC] polish colossalai/nn/lr_scheduler/onecycle.py code style (#1269)	2 years ago