Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
f04cce25
Unverified
Commit
f04cce25
authored
Dec 12, 2023
by
Rhett Ying
Committed by
GitHub
Dec 12, 2023
Browse files
[GraphBolt] rename FromCSC as Create in c++ API (#6730)
parent
1a8e0c00
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
4 additions
and
4 deletions
+4
-4
graphbolt/include/graphbolt/fused_csc_sampling_graph.h
graphbolt/include/graphbolt/fused_csc_sampling_graph.h
+2
-2
graphbolt/src/fused_csc_sampling_graph.cc
graphbolt/src/fused_csc_sampling_graph.cc
+1
-1
graphbolt/src/python_binding.cc
graphbolt/src/python_binding.cc
+1
-1
No files found.
graphbolt/include/graphbolt/fused_csc_sampling_graph.h
View file @
f04cce25
...
...
@@ -78,7 +78,7 @@ class FusedCSCSamplingGraph : public torch::CustomClassHolder {
const
torch
::
optional
<
EdgeAttrMap
>&
edge_attributes
);
/**
* @brief Create a
homogeneous
CSC graph from tensors of CSC format.
* @brief Create a
fused
CSC graph from tensors of CSC format.
* @param indptr Index pointer array of the CSC.
* @param indices Indices array of the CSC.
* @param node_type_offset A tensor representing the offset of node types, if
...
...
@@ -93,7 +93,7 @@ class FusedCSCSamplingGraph : public torch::CustomClassHolder {
*
* @return FusedCSCSamplingGraph
*/
static
c10
::
intrusive_ptr
<
FusedCSCSamplingGraph
>
FromCSC
(
static
c10
::
intrusive_ptr
<
FusedCSCSamplingGraph
>
Create
(
const
torch
::
Tensor
&
indptr
,
const
torch
::
Tensor
&
indices
,
const
torch
::
optional
<
torch
::
Tensor
>&
node_type_offset
,
const
torch
::
optional
<
torch
::
Tensor
>&
type_per_edge
,
...
...
graphbolt/src/fused_csc_sampling_graph.cc
View file @
f04cce25
...
...
@@ -69,7 +69,7 @@ FusedCSCSamplingGraph::FusedCSCSamplingGraph(
TORCH_CHECK
(
indptr
.
device
()
==
indices
.
device
());
}
c10
::
intrusive_ptr
<
FusedCSCSamplingGraph
>
FusedCSCSamplingGraph
::
FromCSC
(
c10
::
intrusive_ptr
<
FusedCSCSamplingGraph
>
FusedCSCSamplingGraph
::
Create
(
const
torch
::
Tensor
&
indptr
,
const
torch
::
Tensor
&
indices
,
const
torch
::
optional
<
torch
::
Tensor
>&
node_type_offset
,
const
torch
::
optional
<
torch
::
Tensor
>&
type_per_edge
,
...
...
graphbolt/src/python_binding.cc
View file @
f04cce25
...
...
@@ -65,7 +65,7 @@ TORCH_LIBRARY(graphbolt, m) {
g
->
SetState
(
state
);
return
g
;
});
m
.
def
(
"from_fused_csc"
,
&
FusedCSCSamplingGraph
::
FromCSC
);
m
.
def
(
"from_fused_csc"
,
&
FusedCSCSamplingGraph
::
Create
);
m
.
def
(
"load_from_shared_memory"
,
&
FusedCSCSamplingGraph
::
LoadFromSharedMemory
);
m
.
def
(
"unique_and_compact"
,
&
UniqueAndCompact
);
...
...
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