Commit 2f610df6 authored by Yuxin Wu's avatar Yuxin Wu

use abstract sockets on linux (#455)

parent ea623424
# -*- coding: UTF-8 -*-
# File: parallel.py
import sys
import weakref
from contextlib import contextmanager
import multiprocessing as mp
......@@ -42,6 +42,13 @@ def _bind_guard(sock, name):
def _get_pipe_name(name):
if sys.platform.startswith('linux'):
# linux supports abstract sockets: http://api.zeromq.org/4-1:zmq-ipc
pipename = "ipc://@{}-pipe-{}".format(name, str(uuid.uuid1())[:8])
pipedir = os.environ.get('TENSORPACK_PIPEDIR', None)
if pipedir is not None:
logger.warn("TENSORPACK_PIPEDIR is not used on Linux any more! Abstract sockets will be used.")
else:
pipedir = os.environ.get('TENSORPACK_PIPEDIR', None)
if pipedir is not None:
logger.info("ZMQ uses TENSORPACK_PIPEDIR={}".format(pipedir))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment