Shortcuts

Source code for torch.distributed.elastic.multiprocessing.subprocess_handler.handlers

#!/usr/bin/env python3
# mypy: allow-untyped-defs

# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
from typing import Dict, Tuple

from torch.distributed.elastic.multiprocessing.subprocess_handler.subprocess_handler import (
    SubprocessHandler,
)

__all__ = ["get_subprocess_handler"]


[docs]def get_subprocess_handler( entrypoint: str, args: Tuple, env: Dict[str, str], stdout: str, stderr: str, local_rank_id: int, ): return SubprocessHandler( entrypoint=entrypoint, args=args, env=env, stdout=stdout, stderr=stderr, local_rank_id=local_rank_id, )

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources