bugfix in terminate function

alicevision · Dec 23, 2024 · 0fc2633 · 0fc2633
1 parent 4a422f5
commit 0fc2633
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/aliceVision/segmentation/segmentation.cpp b/src/aliceVision/segmentation/segmentation.cpp
@@ -89,11 +89,14 @@ bool Segmentation::initialize()
 
 bool Segmentation::terminate()
 {
+    if (_parameters.useGpu)
+    {
 #if ALICEVISION_IS_DEFINED(ALICEVISION_HAVE_ONNX_GPU)
-    Ort::MemoryInfo mem_info_cuda("Cuda", OrtAllocatorType::OrtArenaAllocator, 0, OrtMemType::OrtMemTypeDefault);
-    Ort::Allocator cudaAllocator(*_ortSession, mem_info_cuda);
-    cudaAllocator.Free(_cudaInput);
+        Ort::MemoryInfo mem_info_cuda("Cuda", OrtAllocatorType::OrtArenaAllocator, 0, OrtMemType::OrtMemTypeDefault);
+        Ort::Allocator cudaAllocator(*_ortSession, mem_info_cuda);
+        cudaAllocator.Free(_cudaInput);
 #endif
+    }
 
     return true;
 }