Struct DistributedGNN 

pub struct DistributedGNN {
    pub config: DistributedConfig,
    pub local_partition: GraphData,
    pub partition_info: PartitionInfo,
    pub comm_manager: CommunicationManager,
    pub sync_state: Arc<Mutex<SyncState>>,
    pub metrics: DistributedMetrics,
}

Distributed graph neural network coordinator

Fields

config: DistributedConfig

Configuration

local_partition: GraphData

Local graph partition

partition_info: PartitionInfo

Partition information

comm_manager: CommunicationManager

Communication manager

sync_state: Arc<Mutex<SyncState>>

Parameter synchronization state

metrics: DistributedMetrics

Performance metrics

Implementations

impl DistributedGNN

pub fn new( config: DistributedConfig, full_graph: &GraphData, ) -> Result<Self, DistributedError>

Create a new distributed GNN

pub fn distributed_forward( &mut self, layer: &dyn GraphLayer, ) -> Result<GraphData, DistributedError>

Perform distributed forward pass

pub fn synchronize_parameters( &mut self, parameters: &[Tensor], ) -> Result<Vec<Tensor>, DistributedError>

Synchronize parameters across workers

Trait Implementations

impl Debug for DistributedGNN

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.