Remove cuda output allocator

alicevision · Dec 2, 2024 · c1cbeb6 · c1cbeb6
1 parent 5a04e13
commit c1cbeb6
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/aliceVision/segmentation/segmentation.cpp b/src/aliceVision/segmentation/segmentation.cpp
@@ -72,7 +72,6 @@ bool Segmentation::initialize()
 
         _output.resize(_parameters.classes.size() * _parameters.modelHeight * _parameters.modelWidth);
         _cudaInput = cudaAllocator.Alloc(_output.size() * sizeof(float));
-        _cudaOutput = cudaAllocator.Alloc(_output.size() * sizeof(float));
 #endif
     }
     else
@@ -94,7 +93,6 @@ bool Segmentation::terminate()
     Ort::MemoryInfo mem_info_cuda("Cuda", OrtAllocatorType::OrtArenaAllocator, 0, OrtMemType::OrtMemTypeDefault);
     Ort::Allocator cudaAllocator(*_ortSession, mem_info_cuda);
     cudaAllocator.Free(_cudaInput);
-    cudaAllocator.Free(_cudaOutput);
 #endif
 
     return true;