We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f3983b3 commit cd7dbf2Copy full SHA for cd7dbf2
csrc/cpu/knn_cpu.h
@@ -3,7 +3,6 @@
3
#include <torch/extension.h>
4
#include "utils/neighbors.cpp"
5
#include <iostream>
6
-#include "compat.h"
7
8
torch::Tensor knn_cpu(torch::Tensor support, torch::Tensor query,
9
int64_t k, int64_t n_threads);
csrc/cpu/radius_cpu.h
torch::Tensor radius_cpu(torch::Tensor query, torch::Tensor support,
double radius, int64_t max_num, int64_t n_threads);
0 commit comments