Possible fix for ANN

rapidsai · Jan 26, 2021 · 6004bd9 · 6004bd9
1 parent 6b5e7ff
commit 6004bd9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/cpp/include/cuml/neighbors/knn.hpp b/cpp/include/cuml/neighbors/knn.hpp
@@ -46,6 +46,10 @@ struct knnIndex {
   ~knnIndex() {
     delete index;
     delete gpu_res;
+
+    // to check that destructor is not called twice
+    index = nullptr;
+    gpu_res = nullptr;
   }
 };
 

diff --git a/cpp/src_prims/selection/knn.cuh b/cpp/src_prims/selection/knn.cuh
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2019-2020, NVIDIA CORPORATION.
+ * Copyright (c) 2019-2021, NVIDIA CORPORATION.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -270,7 +270,8 @@ void approx_knn_build_index(ML::knnIndex *index, ML::knnIndexParam *params,
 
   faiss::gpu::StandardGpuResources *gpu_res =
     new faiss::gpu::StandardGpuResources();
-  gpu_res->noTempMemory();
+  //gpu_res->noTempMemory();
+  gpu_res->setTempMemory(3000000000);
   gpu_res->setCudaMallocWarning(false);
   gpu_res->setDefaultStream(device, userStream);
   index->gpu_res = gpu_res;