kaolin
kaolin copied to clipboard
Memory Efficient SPC Trilinear Interpolation
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.