Skip to content

Commit 9d35b05

Browse files
committed
Remove cuda output allocator
1 parent fba5883 commit 9d35b05

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

src/aliceVision/segmentation/segmentation.cpp

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,6 @@ bool Segmentation::initialize()
7272

7373
_output.resize(_parameters.classes.size() * _parameters.modelHeight * _parameters.modelWidth);
7474
_cudaInput = cudaAllocator.Alloc(_output.size() * sizeof(float));
75-
_cudaOutput = cudaAllocator.Alloc(_output.size() * sizeof(float));
7675
#endif
7776
}
7877
else
@@ -94,7 +93,6 @@ bool Segmentation::terminate()
9493
Ort::MemoryInfo mem_info_cuda("Cuda", OrtAllocatorType::OrtArenaAllocator, 0, OrtMemType::OrtMemTypeDefault);
9594
Ort::Allocator cudaAllocator(*_ortSession, mem_info_cuda);
9695
cudaAllocator.Free(_cudaInput);
97-
cudaAllocator.Free(_cudaOutput);
9896
#endif
9997

10098
return true;

0 commit comments

Comments
 (0)