@karmus89 my code doesn’t have any device-specific instructions. All relevant tensors are stored on a single CUDA device.
If there is a slow-down in my code, it’s likely due to either using a lot of indexing, or calling clone(). Any chance cloning could show up as to in the profiler?
Thanks.
Note: I’m not explicitly calling to anywhere in my code