Merge 7a6c1d5 into 50ba399

rapidsai · Oct 21, 2022 · d42d1a6 · d42d1a6
2 parents 50ba399 + 7a6c1d5
commit d42d1a6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/cpp/src/c_api/uniform_neighbor_sampling.cpp b/cpp/src/c_api/uniform_neighbor_sampling.cpp
@@ -310,5 +310,6 @@ extern "C" void cugraph_sample_result_free(cugraph_sample_result_t* result)
   delete internal_pointer->label_;
   delete internal_pointer->index_;
   delete internal_pointer->count_;
+  delete internal_pointer->experimental_count_;
   delete internal_pointer;
 }
diff --git a/python/pylibcugraph/pylibcugraph/tests/test_neighborhood_sampling.py b/python/pylibcugraph/pylibcugraph/tests/test_neighborhood_sampling.py
@@ -248,11 +248,7 @@ def test_neighborhood_sampling_large_sg_graph(gpubenchmark):
     expected_delta = free_memory_before - free_before_cleanup
     leak = expected_delta - actual_delta
     print(f"  {result_size=} {actual_delta=} {expected_delta=} {leak=}")
-    # FIXME: this assertion is commented out until the memory leak is
-    # found. This should be the only failing assertion, so commenting it out
-    # will allow CI to make any other failures more noticeable.
-    #
-    # assert free_memory_before == device.mem_info[0]
+    assert free_memory_before == device.mem_info[0]
 
 
 def test_sample_result():