Unverified Commit f04cce25 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[GraphBolt] rename FromCSC as Create in c++ API (#6730)

parent 1a8e0c00
...@@ -78,7 +78,7 @@ class FusedCSCSamplingGraph : public torch::CustomClassHolder { ...@@ -78,7 +78,7 @@ class FusedCSCSamplingGraph : public torch::CustomClassHolder {
const torch::optional<EdgeAttrMap>& edge_attributes); 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 indptr Index pointer array of the CSC.
* @param indices Indices array of the CSC. * @param indices Indices array of the CSC.
* @param node_type_offset A tensor representing the offset of node types, if * @param node_type_offset A tensor representing the offset of node types, if
...@@ -93,7 +93,7 @@ class FusedCSCSamplingGraph : public torch::CustomClassHolder { ...@@ -93,7 +93,7 @@ class FusedCSCSamplingGraph : public torch::CustomClassHolder {
* *
* @return FusedCSCSamplingGraph * @return FusedCSCSamplingGraph
*/ */
static c10::intrusive_ptr<FusedCSCSamplingGraph> FromCSC( static c10::intrusive_ptr<FusedCSCSamplingGraph> Create(
const torch::Tensor& indptr, const torch::Tensor& indices, const torch::Tensor& indptr, const torch::Tensor& indices,
const torch::optional<torch::Tensor>& node_type_offset, const torch::optional<torch::Tensor>& node_type_offset,
const torch::optional<torch::Tensor>& type_per_edge, const torch::optional<torch::Tensor>& type_per_edge,
......
...@@ -69,7 +69,7 @@ FusedCSCSamplingGraph::FusedCSCSamplingGraph( ...@@ -69,7 +69,7 @@ FusedCSCSamplingGraph::FusedCSCSamplingGraph(
TORCH_CHECK(indptr.device() == indices.device()); 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::Tensor& indptr, const torch::Tensor& indices,
const torch::optional<torch::Tensor>& node_type_offset, const torch::optional<torch::Tensor>& node_type_offset,
const torch::optional<torch::Tensor>& type_per_edge, const torch::optional<torch::Tensor>& type_per_edge,
......
...@@ -65,7 +65,7 @@ TORCH_LIBRARY(graphbolt, m) { ...@@ -65,7 +65,7 @@ TORCH_LIBRARY(graphbolt, m) {
g->SetState(state); g->SetState(state);
return g; return g;
}); });
m.def("from_fused_csc", &FusedCSCSamplingGraph::FromCSC); m.def("from_fused_csc", &FusedCSCSamplingGraph::Create);
m.def( m.def(
"load_from_shared_memory", &FusedCSCSamplingGraph::LoadFromSharedMemory); "load_from_shared_memory", &FusedCSCSamplingGraph::LoadFromSharedMemory);
m.def("unique_and_compact", &UniqueAndCompact); m.def("unique_and_compact", &UniqueAndCompact);
......
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