Skip to content

GitLab

Unverified Commit ec2d458d authored by Rick Ho's avatar Rick Ho Committed by GitHub
Browse files

Merge pull request #51 from laekov/fix-expert-exchange 

use single variable for returned value
parents 28bfe689 318513ae
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
fmoe/gates/utils.py
View file @ ec2d458d
...@@ -16,7 +16,7 @@ def limit_by_capacity(topk_idx, num_expert, world_size, capacity): ...@@ -16,7 +16,7 @@ def limit_by_capacity(topk_idx, num_expert, world_size, capacity):
new_gec = fmoe_native.limit_by_capacity(gec, capacity, new_gec = fmoe_native.limit_by_capacity(gec, capacity,
num_expert, world_size) num_expert, world_size)
if world_size > 1: if world_size > 1:
new_lec, = fmoe_native.expert_exchange(new_gec, num_expert, new_lec = fmoe_native.expert_exchange(new_gec, num_expert,
world_size) world_size)
else: else:
new_lec = new_gec new_lec = new_gec
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment