"docs/source/en/optimization/onnx.md" did not exist on "2a69c0b7b8c4ab2ced90a1784a3eeef6ddf5ae8c"
-
Justin Johnson authored
Summary: We have multiple KNN CUDA implementations. From python, users can currently request a particular implementation via the `version` flag, but they have no way of knowing which implementations can be used for a given problem. This diff exposes a function `pytorch3d._C.knn_check_version(version, D, K)` that returns whether a particular version can be used. Reviewed By: nikhilaravi Differential Revision: D21162573 fbshipit-source-id: 6061960bdcecba454fd920b00036f4e9ff3fdbc0
9f31a4fd
