Function PYBIND11_MODULE

graphlearn/python/c/py_export.cc:35–258

#endif

PYBIND11_MODULE(pywrap_graphlearn, m) {
  m.doc() = "Python interface for graph-learn.";
  // setters
  m.def("set_default_neighbor_id", &SetGlobalFlagDefaultNeighborId);
  m.def("set_tracker_mode", &SetGlobalFlagTrackerMode);
  m.def("set_padding_mode", &SetGlobalFlagPaddingMode);
  m.def("set_storage_mode", &SetGlobalFlagStorageMode);
  m.def("set_default_int_attr", &SetGlobalFlagDefaultIntAttribute);
  m.def("set_default_float_attr", &SetGlobalFlagDefaultFloatAttribute);
  m.def("set_default_string_attr", &SetGlobalFlagDefaultStringAttribute);
  m.def("set_retry_times", &SetGlobalFlagRetryTimes);
  m.def("set_timeout", &SetGlobalFlagTimeout);
  m.def("set_inmemory_queuesize", &SetGlobalFlagInMemoryQueueSize);
  m.def("set_inner_threadnum", &SetGlobalFlagInterThreadNum);
  m.def("set_inter_threadnum", &SetGlobalFlagInterThreadNum);
  m.def("set_intra_threadnum", &SetGlobalFlagIntraThreadNum);
  m.def("set_datainit_batchsize", &SetGlobalFlagDataInitBatchSize);
  m.def("set_shuffle_buffer_size", &SetGlobalFlagShuffleBufferSize);
  m.def("set_rpc_message_max_size", &SetGlobalFlagRpcMessageMaxSize);
  m.def("set_deploy_mode", &SetGlobalFlagDeployMode);
  m.def("set_client_id", &SetGlobalFlagClientId);
  m.def("set_client_count", &SetGlobalFlagClientCount);
  m.def("set_server_count", &SetGlobalFlagServerCount);
  m.def("set_tracker", &SetGlobalFlagTracker);
  m.def("set_server_hosts", &SetGlobalFlagServerHosts);
  m.def("set_knn_metric", &SetGlobalFlagKnnMetric);
  m.def("set_tape_capacity", &SetGlobalFlagTapeCapacity);
  m.def("set_dataset_capacity", &SetGlobalFlagDatasetCapacity);
  m.def("set_ignore_invalid", &SetGlobalFlagIgnoreInvalid);
  m.def("set_neg_sampler_retry_times", &SetGlobalFlagNegativeSamplingRetryTimes);
  m.def("set_field_delimiter", &SetGlobalFlagFieldDelimiter);
  // For Actor
  m.def("set_enable_actor", &SetGlobalFlagEnableActor);
  m.def("set_actor_local_shard_count", &SetGlobalFlagActorLocalShardCount);

  // Constants
  m.attr("kPartitionKey") = kPartitionKey;
  m.attr("kOpName") = kOpName;
  m.attr("kNodeType") = kNodeType;
  m.attr("kEdgeType") = kEdgeType;
  m.attr("kType") = kType;
  m.attr("kSrcType") = kSrcType;
  m.attr("kDstType") = kDstType;
  m.attr("kSrcIds") = kSrcIds;
  m.attr("kDstIds") = kDstIds;
  m.attr("kNodeIds") = kNodeIds;
  m.attr("kEdgeIds") = kEdgeIds;
  m.attr("kNeighborCount") = kNeighborCount;
  m.attr("kNeighborIds") = kNeighborIds;
  m.attr("kBatchSize") = kBatchSize;
  m.attr("kIsSparse") = kIsSparse;
  m.attr("kStrategy") = kStrategy;
  m.attr("kDegreeKey") = kDegreeKey;
  m.attr("kWeightKey") = kWeightKey;
  m.attr("kLabelKey") = kLabelKey;
  m.attr("kIntAttrKey") = kIntAttrKey;
  m.attr("kFloatAttrKey") = kFloatAttrKey;
  m.attr("kStringAttrKey") = kStringAttrKey;