Commit 7302be6b by Wenjie Huang

move ldg_partition from csrc to third_party

parent 542831ac
......@@ -166,8 +166,8 @@ cython_debug/
*.pt
/nohup.out
/cora
/dataset
/test_*
/*.ipynb
/third_party
!/third_party/ldg_partition
[submodule "csrc/partition/neighbor_clustering"]
path = csrc/partition/neighbor_clustering
[submodule "third_party/ldg_partition"]
path = third_party/ldg_partition
url = https://gitee.com/onlynagesha/graph-partition-v4
......@@ -82,7 +82,7 @@ endif()
if (WITH_LDG)
# Imports neighbor-clustering based (e.g. LDG algorithm) graph partitioning implementation
add_definitions(-DWITH_LDG)
set(LDG_DIR "csrc/partition/neighbor_clustering")
set(LDG_DIR "third_party/ldg_partition")
add_library(ldg_partition SHARED "csrc/partition/ldg.cpp")
target_link_libraries(ldg_partition PRIVATE ${TORCH_LIBRARIES})
......
#include <torch/all.h>
#include "neighbor_clustering/vertex_partition/vertex_partition.h"
#include "neighbor_clustering/vertex_partition/params.h"
#include "vertex_partition/vertex_partition.h"
#include "vertex_partition/params.h"
at::Tensor ldg_partition(at::Tensor edges,
at::optional<at::Tensor> vertex_weights,
......
neighbor_clustering @ 5e8dd2eb
Subproject commit 5e8dd2ebda4b0e3b3dec8f43c8ced67785646eef
ldg_partition @ da31e8bb
Subproject commit da31e8bb74c296821a3d0c88279c1d3400b76f03
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment