how to solve it ? I have tried tensor.to(A.device) and tensor.to(accelerator.device), but it seems not to work.
Hi @weizhenhuan , can you provide a reproducer ?