Make the code neutral to device by removing `.cuda()` - Githubissues

laekov / fastmoe

A fast MoE impl for PyTorch

https://fastmoe.ai

Apache License 2.0

1.56k stars 188 forks source link

Make the code neutral to device by removing `.cuda()` #205

Closed roastduck closed 5 months ago