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

use single variable for returned value

Merge pull request #51 from laekov/fix-expert-exchange
use single variable for returned value
ec2d458d · Rick Ho · GitHub · 28bfe689 · 318513ae · ec2d458d
Unverified Commit ec2d458d authored Jun 17, 2021 by Rick Ho Committed by GitHub Jun 17, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

fmoe/gates/utils.py fmoe/gates/utils.py +1 -1

No files found.
--- a/fmoe/gates/utils.py
+++ b/fmoe/gates/utils.py
@@ -16,7 +16,7 @@ def limit_by_capacity(topk_idx, num_expert, world_size, capacity):
        new_gec = fmoe_native.limit_by_capacity(gec, capacity,
                num_expert, world_size)
        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)
        else:
            new_lec = new_gec