gigl.distributed.graph_store.storage_main#

Built-in GiGL Graph Store Server.

Derivved from alibaba/graphlearn-for-pytorch

Attributes#

Functions#

storage_node_process(storage_rank, cluster_info, ...)

Run a storage node process

Module Contents#

gigl.distributed.graph_store.storage_main.storage_node_process(storage_rank, cluster_info, task_config_uri, is_inference, tf_record_uri_pattern='.*-of-.*\\.tfrecord(\\.gz)?$')[source]#

Run a storage node process

Should be called once per storage node (machine).

Parameters:
  • storage_rank (int) – The rank of the storage node.

  • cluster_info (GraphStoreInfo) – The cluster information.

  • task_config_uri (Uri) – The task config URI.

  • is_inference (bool) – Whether the process is an inference process.

  • tf_record_uri_pattern (str) – The TF Record URI pattern.

Return type:

None

gigl.distributed.graph_store.storage_main.logger[source]#
gigl.distributed.graph_store.storage_main.parser[source]#