Remove merge conflict

cpp/tests/community/ecg_test.cpp

@@ -0,0 +1,149 @@
+/*
+ * Copyright (c) 2019-2024, NVIDIA CORPORATION.  All rights reserved.
+ *
+ * NVIDIA CORPORATION and its licensors retain all intellectual property
+ * and proprietary rights in and to this software, related documentation
+ * and any modifications thereto.  Any use, reproduction, disclosure or
+ * distribution of this software and related documentation without an express
+ * license agreement from NVIDIA CORPORATION is strictly prohibited.
+ *
+ */
+#include <utilities/base_fixture.hpp>
+
+#include <cugraph/algorithms.hpp>
+#include <cugraph/legacy/graph.hpp>
+
+#include <rmm/device_vector.hpp>
+
+#include <thrust/host_vector.h>
+
+// FIXME:  Temporarily disable this test.  Something is wrong with
+//         ECG, or the expectation of this test.  If I run ensemble size
+//         of 24 this fails.  It also fails with the SG Louvain change
+//         for PR 1271
+#if 0
+TEST(legacy_ecg, success)
+{
+  // FIXME: verify that this is the karate dataset
+  std::vector<int> off_h = {0,  16,  25,  35,  41,  44,  48,  52,  56,  61,  63, 66,
+                            67, 69,  74,  76,  78,  80,  82,  84,  87,  89,  91, 93,
+                            98, 101, 104, 106, 110, 113, 117, 121, 127, 139, 156};
+  std::vector<int> ind_h = {
+    1,  2,  3,  4,  5,  6,  7,  8,  10, 11, 12, 13, 17, 19, 21, 31, 0,  2,  3,  7,  13, 17, 19,
+    21, 30, 0,  1,  3,  7,  8,  9,  13, 27, 28, 32, 0,  1,  2,  7,  12, 13, 0,  6,  10, 0,  6,
+    10, 16, 0,  4,  5,  16, 0,  1,  2,  3,  0,  2,  30, 32, 33, 2,  33, 0,  4,  5,  0,  0,  3,
+    0,  1,  2,  3,  33, 32, 33, 32, 33, 5,  6,  0,  1,  32, 33, 0,  1,  33, 32, 33, 0,  1,  32,
+    33, 25, 27, 29, 32, 33, 25, 27, 31, 23, 24, 31, 29, 33, 2,  23, 24, 33, 2,  31, 33, 23, 26,
+    32, 33, 1,  8,  32, 33, 0,  24, 25, 28, 32, 33, 2,  8,  14, 15, 18, 20, 22, 23, 29, 30, 31,
+    33, 8,  9,  13, 14, 15, 18, 19, 20, 22, 23, 26, 27, 28, 29, 30, 31, 32};
+
+  std::vector<float> w_h(ind_h.size(), float{1.0});
+
+  int num_verts = off_h.size() - 1;
+  int num_edges = ind_h.size();
+
+  thrust::host_vector<int> cluster_id(num_verts, -1);
+
+  rmm::device_vector<int> offsets_v(off_h);
+  rmm::device_vector<int> indices_v(ind_h);
+  rmm::device_vector<float> weights_v(w_h);
+  rmm::device_vector<int> result_v(cluster_id);
+
+  cugraph::legacy::GraphCSRView<int, int, float> graph_csr(
+    offsets_v.data().get(), indices_v.data().get(), weights_v.data().get(), num_verts, num_edges);
+
+  raft::handle_t handle;
+  cugraph::ecg<int32_t, int32_t, float>(handle, graph_csr, .05, 16, result_v.data().get());
+
+  cluster_id = result_v;
+  int max    = *max_element(cluster_id.begin(), cluster_id.end());
+  int min    = *min_element(cluster_id.begin(), cluster_id.end());
+
+  ASSERT_EQ((min >= 0), 1);
+
+  std::set<int> cluster_ids;
+  for (auto c : cluster_id) { cluster_ids.insert(c); }
+
+  ASSERT_EQ(cluster_ids.size(), size_t(max + 1));
+
+  float modularity{0.0};
+
+  cugraph::ext_raft::analyzeClustering_modularity(
+    graph_csr, max + 1, result_v.data().get(), &modularity);
+
+  // 0.399 is 5% below the reference value returned in
+  // <cugraph>/python/utils/ECG_Golden.ipynb on the same dataset
+  ASSERT_GT(modularity, 0.399);
+}
+#endif
+
+TEST(legacy_ecg, dolphin)
+{
+  raft::handle_t handle;
+
+  auto stream = handle.get_stream();
+
+  std::vector<int> off_h = {0,   6,   14,  18,  21,  22,  26,  32,  37,  43,  50,  55,  56,
+                            57,  65,  77,  84,  90,  99,  106, 110, 119, 125, 126, 129, 135,
+                            138, 141, 146, 151, 160, 165, 166, 169, 179, 184, 185, 192, 203,
+                            211, 213, 221, 226, 232, 239, 243, 254, 256, 262, 263, 265, 272,
+                            282, 286, 288, 295, 297, 299, 308, 309, 314, 315, 318};
+  std::vector<int> ind_h = {
+    10, 14, 15, 40, 42, 47, 17, 19, 26, 27, 28, 36, 41, 54, 10, 42, 44, 61, 8,  14, 59, 51, 9,
+    13, 56, 57, 9,  13, 17, 54, 56, 57, 19, 27, 30, 40, 54, 3,  20, 28, 37, 45, 59, 5,  6,  13,
+    17, 32, 41, 57, 0,  2,  29, 42, 47, 51, 33, 5,  6,  9,  17, 32, 41, 54, 57, 0,  3,  16, 24,
+    33, 34, 37, 38, 40, 43, 50, 52, 0,  18, 24, 40, 45, 55, 59, 14, 20, 33, 37, 38, 50, 1,  6,
+    9,  13, 22, 25, 27, 31, 57, 15, 20, 21, 24, 29, 45, 51, 1,  7,  30, 54, 8,  16, 18, 28, 36,
+    38, 44, 47, 50, 18, 29, 33, 37, 45, 51, 17, 36, 45, 51, 14, 15, 18, 29, 45, 51, 17, 26, 27,
+    1,  25, 27, 1,  7,  17, 25, 26, 1,  8,  20, 30, 47, 10, 18, 21, 24, 35, 43, 45, 51, 52, 7,
+    19, 28, 42, 47, 17, 9,  13, 60, 12, 14, 16, 21, 34, 37, 38, 40, 43, 50, 14, 33, 37, 44, 49,
+    29, 1,  20, 23, 37, 39, 40, 59, 8,  14, 16, 21, 33, 34, 36, 40, 43, 45, 61, 14, 16, 20, 33,
+    43, 44, 52, 58, 36, 57, 0,  7,  14, 15, 33, 36, 37, 52, 1,  9,  13, 54, 57, 0,  2,  10, 30,
+    47, 50, 14, 29, 33, 37, 38, 46, 53, 2,  20, 34, 38, 8,  15, 18, 21, 23, 24, 29, 37, 50, 51,
+    59, 43, 49, 0,  10, 20, 28, 30, 42, 57, 34, 46, 14, 16, 20, 33, 42, 45, 51, 4,  11, 18, 21,
+    23, 24, 29, 45, 50, 55, 14, 29, 38, 40, 43, 61, 1,  6,  7,  13, 19, 41, 57, 15, 51, 5,  6,
+    5,  6,  9,  13, 17, 39, 41, 48, 54, 38, 3,  8,  15, 36, 45, 32, 2,  37, 53};
+
+  std::vector<float> w_h(ind_h.size(), float{1.0});
+
+  int num_verts = off_h.size() - 1;
+  int num_edges = ind_h.size();
+
+  rmm::device_uvector<int> offsets_v(num_verts + 1, stream);
+  rmm::device_uvector<int> indices_v(num_edges, stream);
+  rmm::device_uvector<float> weights_v(num_edges, stream);
+  rmm::device_uvector<int> result_v(num_verts, stream);
+
+  raft::update_device(offsets_v.data(), off_h.data(), off_h.size(), stream);
+  raft::update_device(indices_v.data(), ind_h.data(), ind_h.size(), stream);
+  raft::update_device(weights_v.data(), w_h.data(), w_h.size(), stream);
+
+  cugraph::legacy::GraphCSRView<int, int, float> graph_csr(
+    offsets_v.data(), indices_v.data(), weights_v.data(), num_verts, num_edges);
+
+  cugraph::ecg<int32_t, int32_t, float>(handle, graph_csr, .05, 16, result_v.data());
+
+  auto cluster_id = cugraph::test::to_host(handle, result_v);
+
+  int max = *max_element(cluster_id.begin(), cluster_id.end());
+  int min = *min_element(cluster_id.begin(), cluster_id.end());
+
+  ASSERT_EQ((min >= 0), 1);
+
+  std::set<int> cluster_ids;
+  for (auto c : cluster_id) {
+    cluster_ids.insert(c);
+  }
+
+  ASSERT_EQ(cluster_ids.size(), size_t(max + 1));
+
+  float modularity{0.0};
+
+  cugraph::ext_raft::analyzeClustering_modularity(graph_csr, max + 1, result_v.data(), &modularity);
+
+  float random_modularity{0.95 * 0.4962422251701355};
+
+  ASSERT_GT(modularity, random_modularity);
+}
+
+CUGRAPH_TEST_PROGRAM_MAIN()