add platform limit for cleanup function, test=develop (#22846)

revert-22710-feature/integrated_ps_api
Chen Weihang 5 years ago committed by GitHub
parent 1a533ed2de
commit 2e9e8c056a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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):

Loading…
Cancel
Save