From 92f919fe11cac49a9449c74e097f069d0fd78797 Mon Sep 17 00:00:00 2001
From: Vincent Demoulin <vincent.demoulin@gmail.com>
Date: Thu, 21 Nov 2024 18:57:49 +0100
Subject: [PATCH] Remove cuda output allocator

---
 src/aliceVision/segmentation/segmentation.cpp | 2 --
 1 file changed, 2 deletions(-)

diff --git a/src/aliceVision/segmentation/segmentation.cpp b/src/aliceVision/segmentation/segmentation.cpp
index c45ada3491..213b68b023 100644
--- 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;