Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
B
BTS-MTGNN
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
zhlj
BTS-MTGNN
Commits
7302be6b
Commit
7302be6b
authored
Dec 25, 2023
by
Wenjie Huang
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
move ldg_partition from csrc to third_party
parent
542831ac
Show whitespace changes
Inline
Side-by-side
Showing
6 changed files
with
7 additions
and
7 deletions
+7
-7
.gitignore
+1
-1
.gitmodules
+2
-2
CMakeLists.txt
+1
-1
csrc/partition/ldg.cpp
+2
-2
csrc/partition/neighbor_clustering
+0
-1
third_party/ldg_partition
+1
-0
No files found.
.gitignore
View file @
7302be6b
...
...
@@ -166,8 +166,8 @@ cython_debug/
*.pt
/nohup.out
/cora
/dataset
/test_*
/*.ipynb
/third_party
!/third_party/ldg_partition
.gitmodules
View file @
7302be6b
[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
CMakeLists.txt
View file @
7302be6b
...
...
@@ -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
}
)
...
...
csrc/partition/ldg.cpp
View file @
7302be6b
#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
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment