update

3be6ac10 · Jiezhong Qiu · 6212aa28 · 3be6ac10
Commit 3be6ac10 authored Jan 03, 2021 by Jiezhong Qiu
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

pytorch/cuda/moe.py pytorch/cuda/moe.py +0 -1

No files found.
--- a/pytorch/cuda/moe.py
+++ b/pytorch/cuda/moe.py
@@ -133,7 +133,6 @@ def test_dp():
    moe = MOELayer(num_expert, in_feat, out_feat).cuda()
    moe_dp = torch.nn.DataParallel(moe, device_ids=[0,1,2])
    for i in range(5):
-        print(i, "forward")
        output = moe_dp(inp, gate)