nv_dfm_core.api.PreparedPipeline#

class nv_dfm_core.api.PreparedPipeline(api_version, federation_module_name, homesite, pipeline_name, net_irs, yield_places)[source]#

A pipeline that has been prepared and optimized for execution in a federation.

PreparedPipeline contains the intermediate representation (IR) for each participating site, along with metadata needed for execution.

Parameters:

  • api_version (str)

  • federation_module_name (str)

  • homesite (str)

  • pipeline_name (str | None)

  • net_irs (list[Any])

  • yield_places (list[Any])

bind_net_ir(site, input_params)[source]#

Bind a specific site’s NetIR with input parameters to create a BoundNetIR ready for execution.

Parameters:

  • site (str)

  • input_params (list[tuple[Any, dict[str, Any]]])

Return type:

Any

bind_net_irs(input_params)[source]#

Bind all site NetIRs with input parameters to create BoundNetIRs for each participating site.

Parameters:

input_params (list[tuple[Any, dict[str, Any]]])

Return type:

dict[str, Any]

check_callbacks(callback_places, logger)[source]#

Check and warn if callback registration doesn’t match yield places in the pipeline.

Parameters:

  • callback_places (list[str])

  • logger (Logger)

check_input_params(input_params)[source]#

Validate that all required input parameters are provided and no extra parameters are given.

Parameters:

input_params (list[tuple[Any, dict[str, Any]]])

Return type:

None