DickJC123 commented on issue #12662: Memory leak when passing images of different dimensions with MXNET_CUDNN_AUTOTUNE_DEFAULT URL: https://github.com/apache/incubator-mxnet/issues/12662#issuecomment-437541465 As part of #12804, I took the minimal example code provided here and turned it into the initially failing unittest test_gluon_gpu.py:test_large_models. To make the test universal, I had its operation scale with the amount of available memory, as provided by a new Python API addition gpu_memory_info(): ``` python -c "import mxnet; print(mxnet.context.gpu_memory_info(0))" ``` After CI showed the test failing, I pushed the fix that corrected the issue. So in summary, we have a new function to measure memory, and a new unittest for this issue.
---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on 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
