code: def swipe_once(self, idx, capacity, bias):
      with torch.no_grad():
          idx_new, capacity = fmoe_native.swipe_once(
              idx, capacity, self.num_expert, self.world_size, bias
          )
          idx_new = idx_new.to(idx.device)
      return idx_new, capacity
  
file_path: smart-moe/fmoe/gates/swipe_gate.py
imports: import math
  import torch
  import torch.distributed
  import torch.nn
  import torch.nn.functional
  from .naive_gate import NaiveGate
  from fmoe.functions import count_by_gate
  import fmoe_cuda
label_desc: Use image node_id 2 for calling the SwipeGate obj's underlying member method code with example usage: obj.swipe_once(idx, capacity, bias) and returns: idx_new,  capacity