vladoovtcharov commented on a change in pull request #14252: Allow clearing gpu
cache
URL: https://github.com/apache/incubator-mxnet/pull/14252#discussion_r281321701
##########
File path: python/mxnet/context.py
##########
@@ -145,6 +145,11 @@ def default_ctx(cls, val):
cls._default_ctx.value = val
#pylint: enable=no-self-argument
+ def release_all(self):
Review comment:
Good point, yes I'll add some documentation to try and make it clear it
won't release all memory, just unreferenced data. The equivalent in pytorch is
called empty_cache so maybe that would be a better name to use
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
