这是indexloc提供的服务，不要输入任何密码
Skip to content

use explicit multiprocessing context for creating Pipe in subproc.py #1102

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Apr 19, 2024
Merged

use explicit multiprocessing context for creating Pipe in subproc.py #1102

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
b6f021d
use explicit multiprocessing context for creating Pipe in subproc.py
Apr 8, 2024
46e0933
Merge branch 'master' into fix/context-dependent-pipe-in-subproc
maxhuettenrauch Apr 16, 2024
7607cae
Merge branch 'master' into fix/context-dependent-pipe-in-subproc
maxhuettenrauch Apr 18, 2024
a4b6bb4
Merge branch 'master' into fix/context-dependent-pipe-in-subproc
maxhuettenrauch Apr 19, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions tianshou/env/worker/subproc.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import time
from collections import OrderedDict
from collections.abc import Callable
from multiprocessing import Pipe, connection
from multiprocessing import connection
from multiprocessing.context import BaseContext
from typing import Any, Literal

Expand Down Expand Up @@ -149,11 +149,11 @@ def __init__(
share_memory: bool = False,
context: BaseContext | Literal["fork", "spawn"] | None = None,
) -> None:
self.parent_remote, self.child_remote = Pipe()
self.share_memory = share_memory
self.buffer: dict | tuple | ShArray | None = None
if not isinstance(context, BaseContext):
context = multiprocessing.get_context(context)
self.parent_remote, self.child_remote = context.Pipe()
self.share_memory = share_memory
self.buffer: dict | tuple | ShArray | None = None
assert hasattr(context, "Process") # for mypy
if self.share_memory:
dummy = env_fn()
Expand Down