iOS: leaks - Githubissues

fbelderink / flutter_pytorch_mobile

A flutter plugin for pytorch model inference. Supports image models as well as custom models.

https://pub.dev/packages/pytorch_mobile

Other

101 stars 52 forks source link

iOS: leaks #24

Open manmaor opened 1 year ago

manmaor commented 1 year ago

Hi, we experienced some leaks in our iOS app, after some investigation we found two memory leaks:

rawBytes in UIImageExtension (calloc not freed)
normalizedBuffer in UIImageExtension (malloc not freed)

from the Profiler: Profiler image

In the first case, I was able to free the memory at the end of the method. But in the second case, we return normalizedBuffer to use as the input, so I freed the input in PyTorchMobilePlugin.

Hoping for your fast response. Maor.

igala commented 1 year ago

@fynnmaarten Hello Fyn,

Can you please review this?