Orchestration

TorchX

Contains utilities for dispatching jobs using TorchX.

This module defines various utility functions for dispatching batch jobs using the TorchX launcher.

class calisim.orchestration.torchx_bridge.TorchXJobLauncher[source]

Utility wrapper around TorchX job orchestration.

launch(orchestration: OrchestrationModel, args: list[str]) → dict[str, Any][source]

Launch the job.

Parameters:

  • orchestration (OrchestrationModel) – The orchestration data model.

  • args (list[str]) – The script arguments.

calisim.orchestration.torchx_bridge.get_def(orchestration: OrchestrationModel, args: list[str]) → AppDef[source]

Get the application definition.

Parameters:

  • orchestration (OrchestrationModel) – The orchestration data model.

  • args (list[str]) – The script arguments.

Returns:

The application definition.

Return type:

specs.AppDef

calisim.orchestration.torchx_bridge.launch(orchestration: OrchestrationModel, definition: AppDef) → dict[str, Any][source]

Dispatch a TorchX job using a scheduler.

Parameters:

  • orchestration (OrchestrationModel) – The orchestration data model.

  • definition (specs.AppDef) – The application definition.