tglite.Mailbox

class tglite.Mailbox(num_nodes: int, size: int, dim: int, device=None)

A container for node mailbox messages.

__init__(num_nodes: int, size: int, dim: int, device=None)

Methods

__init__(num_nodes, size, dim[, device])

dims()

move_to(device, **kwargs)

reset()

store(nids, mail, mail_ts)

Attributes

device

mail

time