Commit 7302be6b by Wenjie Huang

move ldg_partition from csrc to third_party

parent 542831ac
...@@ -166,8 +166,8 @@ cython_debug/ ...@@ -166,8 +166,8 @@ cython_debug/
*.pt *.pt
/nohup.out /nohup.out
/cora
/dataset /dataset
/test_* /test_*
/*.ipynb /*.ipynb
/third_party /third_party
!/third_party/ldg_partition
[submodule "csrc/partition/neighbor_clustering"] [submodule "third_party/ldg_partition"]
path = csrc/partition/neighbor_clustering path = third_party/ldg_partition
url = https://gitee.com/onlynagesha/graph-partition-v4 url = https://gitee.com/onlynagesha/graph-partition-v4
...@@ -82,7 +82,7 @@ endif() ...@@ -82,7 +82,7 @@ endif()
if (WITH_LDG) if (WITH_LDG)
# Imports neighbor-clustering based (e.g. LDG algorithm) graph partitioning implementation # Imports neighbor-clustering based (e.g. LDG algorithm) graph partitioning implementation
add_definitions(-DWITH_LDG) 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") add_library(ldg_partition SHARED "csrc/partition/ldg.cpp")
target_link_libraries(ldg_partition PRIVATE ${TORCH_LIBRARIES}) target_link_libraries(ldg_partition PRIVATE ${TORCH_LIBRARIES})
......
#include <torch/all.h> #include <torch/all.h>
#include "neighbor_clustering/vertex_partition/vertex_partition.h" #include "vertex_partition/vertex_partition.h"
#include "neighbor_clustering/vertex_partition/params.h" #include "vertex_partition/params.h"
at::Tensor ldg_partition(at::Tensor edges, at::Tensor ldg_partition(at::Tensor edges,
at::optional<at::Tensor> vertex_weights, 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