github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-22 13:41:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update test_shard_deepseek.py

Browse Source
This commit is contained in:
Hanks 2025-05-23 11:16:36 +08:00 committed by GitHub
parent 252efa63fc
commit b7df86848c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_shardformer/test_model/test_shard_deepseek.py
View File

@ -195,6 +195,7 @@ def run_deepseek_test(config: Tuple[int, ...]):
(0, 1, 4, 2, 1), (0, 1, 4, 2, 1),
(0, 1, 1, 4, 1), (0, 1, 1, 4, 1),
# (0, 1, 4, 1, 1), # todo: failed pass, need to be fixed # (0, 1, 4, 1, 1), # todo: failed pass, need to be fixed
(0, 1, 2, 1, 1),
# zero 1: # zero 1:
(1, 2, 1, 1, 2), (1, 2, 1, 1, 2),
(1, 2, 1, 4, 1), (1, 2, 1, 4, 1),