allow specifying device

guanyilun / galilei

A generic function emulator that supports multiple backends

MIT License

1 stars 0 forks source link

Closed guanyilun closed 1 year ago

guanyilun commented 1 year ago

allow specifying device through e.g. device='cuda' in emulate function arguments.