matensemble.logger module

class matensemble.logger.StatusWriter(path, nnodes, cores_per_node, gpus_per_node)[source]

Bases: object

Class to handle updating the status file

Parameters:

  • path (Path)

  • nnodes (int)

  • cores_per_node (int)

  • gpus_per_node (int)

path

the path to the status file

Type:

Path

nnodes

The number of nodes that flux is managing (total_allocation - 1 for flux borker)

Type:

int

cores_per_node

The number of CPU cores that are available on each node

Type:

int

gpus_per_node

The number of GPUs that are available on each node

Type:

int

__init__(path, nnodes, cores_per_node, gpus_per_node)[source]
Parameters:

  • path (Path)

  • nnodes (int)

  • cores_per_node (int)

  • gpus_per_node (int)

Return type:

None

update(pending, running, completed, failed, free_cores, free_gpus)[source]
Return type:

None

Parameters:

  • pending (int)

  • running (int)

  • completed (int)

  • failed (int)

  • free_cores (int)

  • free_gpus (int)