ivy-llc / demos

Demos and Tutorials with Ivy
https://ivy.dev/docs/demos/
19 stars 61 forks source link

perf: moved the model and inputs to gpu:0 before transpilation #61

Closed vedpatwardhan closed 1 year ago

vedpatwardhan commented 1 year ago

they weren't gpu arrays orginally because the default device with native torch code is a cpu and we weren't using any ivy's creation functions which would consider gpu as the default device