kaolin icon indicating copy to clipboard operation
kaolin copied to clipboard

Memory Efficient SPC Trilinear Interpolation

Open tovacinni opened this issue 3 years ago • 0 comments

The current trilinear interpolation function within Kaolin is not memory efficient for the purpose of making the backprop easy with autodiff. This is not a performant approach in terms of both memory bandwidth and usage; we should instead just compute the trilinear interpolation in a forward kernel and have a separate kernel for the backprop.

We also need to figure out how to do 2nd derivatives (I have no idea how PyTorch does this).

We have an implementation of the forward pass in an internal MR, but not the backward pass.

tovacinni avatar Nov 04 '21 21:11 tovacinni