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 {
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,
......
......@@ -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,
......
......@@ -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);
......
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