saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-05 04:17:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix cleanup on exit

Browse source
This commit is contained in:
Ryan Hill 2021-05-07 13:16:17 -07:00
parent 9dfdafe6e0
commit 74b1016d3b
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
java/src/main/java/ai/onnxruntime/OnnxRuntime.java
View file

@ -118,12 +118,12 @@ final class OnnxRuntime {
*
* @param file The file to remove.
*/
private static void cleanUp(File file) {
private static void cleanUp(File file, boolean on_exit_only=false) {
if (!file.exists()) {
return;
}
logger.log(Level.FINE, "Deleting " + file);
if (!file.delete()) {
if (on_exit_only || !file.delete()) {
logger.log(Level.FINE, "Deleting " + file + " on exit");
file.deleteOnExit();
}
@ -213,7 +213,7 @@ final class OnnxRuntime {
logger.log(Level.FINE, "Loaded native library '" + library + "' from resource path");
}
} finally {
if (system_load) cleanUp(tempFile);
cleanUp(tempFile, !system_load);
}
}