tiler icon indicating copy to clipboard operation
tiler copied to clipboard

Published 20 hours ago • verified publisher icon the-lay

[IDEA] Tiling PyTorch tensors

Open the-lay opened this issue 3 years ago • 1 comments

Is your feature request related to a problem? Please describe. Pytorch's API is very numpy-like, but has differences in underlying memory model. It might be interesting to take a look what's happening there and whether it is possible to tile pytorch tensors too.

the-lay avatar Dec 15 '21 14:12 the-lay

I've run into issues with this myself and have needed to take things off the gpu just to tile which isn't ideal - I'll take a look and see what the sticking points are

alisterburt avatar Jan 13 '22 18:01 alisterburt