mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-05 11:02:05 +00:00
[shardformer] support ep for deepseek v3 (#6185)
* [feature] support ep for deepseek v3 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * fix test * [shardformer] fix deepseek v3 init * [lazy] fit lora for lazy init * [example] support npu for deepseek v3 --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
@@ -64,7 +64,10 @@ class ProcessGroupMesh:
|
||||
system resources.
|
||||
"""
|
||||
for group in self._ranks_to_group.values():
|
||||
dist.destroy_process_group(group)
|
||||
try:
|
||||
dist.destroy_process_group(group)
|
||||
except ValueError:
|
||||
pass
|
||||
|
||||
# Manually clear all process groups to save memory
|
||||
gc.collect()
|
||||
|
Reference in New Issue
Block a user