diff --git a/simple-knn/simple_knn.cu b/simple-knn/simple_knn.cu index e72e4c9..880a49b 100644 --- a/simple-knn/simple_knn.cu +++ b/simple-knn/simple_knn.cu @@ -23,6 +23,7 @@ #define __CUDACC__ #include #include +#include namespace cg = cooperative_groups; @@ -218,4 +219,4 @@ void SimpleKNN::knn(int P, float3* points, float* meanDists) boxMeanDist << > > (P, points, indices_sorted.data().get(), boxes.data().get(), meanDists); cudaFree(result); -} \ No newline at end of file +}