github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-18 07:31:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

[release] update version (#5752)

Browse Source 
* [release] update version

* [devops] update compatibility test

* [devops] update compatibility test

* [devops] update compatibility test

* [devops] update compatibility test

* [test] fix ddp plugin test

* [test] fix gptj and rpc test

* [devops] fix cuda ext compatibility

* [inference] fix flash decoding test

* [inference] fix flash decoding test
This commit is contained in:
Hongxin Liu
2024-05-31 19:40:26 +08:00
committed by GitHub
parent 677cbfacf8
commit 68359ed1e1
10 changed files with 19 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_booster/test_plugin/test_torch_ddp_plugin.py
View File

@@ -47,7 +47,7 @@ def check_torch_ddp_plugin():
registry = model_zoo
for name, (model_fn, data_gen_fn, output_transform_fn, _, _) in registry.items():
if name == "dlrm_interactionarch":
if name == "dlrm_interactionarch" or name.startswith("simple_"):
continue
run_fn(model_fn, data_gen_fn, output_transform_fn)
torch.cuda.empty_cache()