github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-08 04:24:47 +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_shardformer/test_model/test_shard_gptj.py
View File

@@ -240,7 +240,6 @@ def run_gptj_3d_test(test_config):
def check_gptj(rank, world_size, port):
disable_existing_loggers()
colossalai.launch(
config={},
rank=rank,
world_size=world_size,
host="localhost",
@@ -253,7 +252,6 @@ def check_gptj(rank, world_size, port):
def check_gptj_3d(rank, world_size, port):
disable_existing_loggers()
colossalai.launch(
config={},
rank=rank,
world_size=world_size,
host="localhost",