Is knn_interpolate jittable?

[w-laura]

Is the function knn_interpolate traceable? Because when I try to load a saved model I get an error telling me that the torch_cluster::knn is unknown and may not be supported by TorchScript.

[rusty1s]

Can you show me a simple script to reproduce? Where do you want to load the saved model?