graphstorm.gsf.initialize

graphstorm.gsf.initialize(ip_config, backend, use_wholegraph=False)

Initialize distributed training and inference context.

Parameters

ip_config: str

File path of ip_config file, e.g., /tmp/ip_list.txt.

backend: str

Torch distributed backend, e.g., gloo or nccl.

use_wholegraph: bool

Whether to use wholegraph for feature transfer.