gigl.src.common.utils.gigl_runtime#
Runtime initialization helpers for GiGL component entrypoints.
Functions#
|
Initialize GiGL runtime environment and metrics for a component. |
Module Contents#
- gigl.src.common.utils.gigl_runtime.initialize_gigl_runtime(applied_task_identifier, task_config_uri, resource_config_uri, service_name, component, cpu_docker_uri=None, cuda_docker_uri=None)[source]#
Initialize GiGL runtime environment and metrics for a component.
- Parameters:
applied_task_identifier (str) – Unique identifier for the GiGL job.
task_config_uri (gigl.common.Uri) – URI to the task config YAML file.
resource_config_uri (gigl.common.Uri) – URI to the resource config YAML file.
service_name (str) – Name of the service, used for metric grouping.
component (gigl.src.common.constants.components.GiGLComponents) – GiGL component being initialized.
cpu_docker_uri (Optional[str]) – CPU source image URI. Defaults to the release CPU image.
cuda_docker_uri (Optional[str]) – CUDA source image URI. Defaults to the release CUDA image.
- Return type:
None