pytorch/torch/csrc/multiprocessing/init.h
Pieter Noordhuis 220ce8046e Binding for prctl(PR_SET_PDEATHSIG) (#14491)
Summary:
If torch.multiprocessing.spawn is used to launch non-daemonic
processes (the default since #14391), the spawned children won't be
automatically terminated when the parent terminates.

On Linux, we can address this by setting PR_SET_PDEATHSIG, which
delivers a configurable signal to child processes when their parent
terminates.

Fixes #14394.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14491

Differential Revision: D13270374

Pulled By: pietern

fbshipit-source-id: 092c9d3c3cea2622c3766b467957bc27a1bd500c
2018-11-29 20:09:19 -08:00

12 lines
187 B
C++

#pragma once
#include <torch/csrc/python_headers.h>
namespace torch {
namespace multiprocessing {
PyMethodDef* python_functions();
} // namespace multiprocessing
} // namespace torch