From 2e9e8c056af0873393cef2426982d989fbfb741b Mon Sep 17 00:00:00 2001 From: Chen Weihang Date: Thu, 5 Mar 2020 10:27:33 +0800 Subject: [PATCH] add platform limit for cleanup function, test=develop (#22846) --- python/paddle/fluid/reader.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/paddle/fluid/reader.py b/python/paddle/fluid/reader.py index ef99b82b02..362b23c297 100644 --- a/python/paddle/fluid/reader.py +++ b/python/paddle/fluid/reader.py @@ -137,7 +137,9 @@ class CleanupFuncRegistrar(): # BlockingQueue) may not be completely released, resulting in the corresponding # memory-mapped file remaining on the disk (/dev/shm), so register this function # to clean up shared memory objects in these two queues before the python interpreter exits. -CleanupFuncRegistrar.register(_cleanup) +# NOTE: Currently multi-process DataLoader only supports Linux platform +if not (sys.platform == 'darwin' or sys.platform == 'win32'): + CleanupFuncRegistrar.register(_cleanup) class DataLoaderBase(object): -- GitLab