You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
(Paste the commands you ran that produced the error.)
What have you tried to solve it?
add a method to delete index memory which created by method ai.djl.pytorch.jni.PyTorchLibrary#torchIndexInit
Environment Info
Please run the command ./gradlew debugEnv from the root directory of DJL (if necessary, clone DJL first). It will output information about your system, environment, and installation that can help us debug your issue. Paste the output of the command below:
PASTE OUTPUT HERE
The text was updated successfully, but these errors were encountered:
Description
PtNDArray.get(NDArray index)和PtNDArray.set(NDIndex index, NDArray value) cause memory leak
Expected Behavior
(what's the expected behavior?)
Error Message
memory leak
How to Reproduce?
try (final NDManager ndManager = NDManager.newBaseManager()) {
while (true) {
try (NDManager manager = ndManager.newSubManager()) {
NDArray a = manager.create(new int[1024 * 1024 * 1]);
final NDArray eq = a.eq(1);
a.get(eq);
}
}
}
Steps to reproduce
(Paste the commands you ran that produced the error.)
What have you tried to solve it?
Environment Info
Please run the command
./gradlew debugEnv
from the root directory of DJL (if necessary, clone DJL first). It will output information about your system, environment, and installation that can help us debug your issue. Paste the output of the command below:The text was updated successfully, but these errors were encountered: