serverless_gpu.asynchronous

Asynchronous execution and monitoring for distributed serverless GPU compute.

This module provides asynchronous execution capabilities for serverless GPU jobs, including:

  • Non-blocking job submission and execution

  • Real-time job monitoring and status updates

  • Environment synchronization across distributed nodes

  • Job lifecycle management and error handling

  • Integration with Databricks workspace for job tracking

Functions

get_calls([function_name, ...])

Returns a list of currently running DistributedFunctionCall instances that (optionally) match the specified function.

print_runs(runs)

Prints the details of all specified DistributedFunctionCall instances.

stop(runs)

Stops all specified DistributedFunctionCall instances.

wait(runs)

Waits for all specified job instances to finish.

Classes

serverless_gpu.asynchronous.get_calls(function_name=None, function_callable=None, start_time_from=None, end_time_to=None, status=None, active_only=True, limit=5)[source]

Returns a list of currently running DistributedFunctionCall instances that (optionally) match the specified function.

Parameters

  • function_name (Optional[str]) – Optional name of the function to filter by

  • function_callable (Optional[Callable]) – Optional callable to get function name from

  • start_time_from (Optional[float]) – Optional start time filter

  • end_time_to (Optional[float]) – Optional end time filter

  • status (Optional[RunStatus]) – Optional RunStatus to filter by

  • active_only (bool) – Whether to only return active runs

  • limit (int) – Optional maximum number of calls to return

serverless_gpu.asynchronous.print_runs(runs)[source]

Prints the details of all specified DistributedFunctionCall instances. If no runs are specified, prints all currently running DistributedFunctionCall instances.

Return type

None

serverless_gpu.asynchronous.stop(runs)[source]

Stops all specified DistributedFunctionCall instances. If no runs are specified, stops all currently running DistributedFunctionCall instances.

Return type

None

serverless_gpu.asynchronous.wait(runs)[source]

Waits for all specified job instances to finish.

Return type

None