Skip to content

Commit

Permalink
comment applied
Browse files Browse the repository at this point in the history
  • Loading branch information
yf711 committed Jun 28, 2024
1 parent dbcc3c7 commit f2c2915
Showing 1 changed file with 0 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -321,7 +321,6 @@ void* OutputAllocator::reallocateOutput(char const* /*tensorName*/, void* /*curr
uint64_t /*alignment*/) noexcept {
// Some memory allocators return nullptr when allocating zero bytes, but TensorRT requires a non-null ptr
// even for empty tensors, so allocate a dummy byte.
LOGS_DEFAULT(WARNING) << "[TensorRT EP] reallocateOutput() is deprecated after TensorRT 8.6";
size = std::max(size, static_cast<uint64_t>(1));
if (size > allocated_size) {
cudaFree(outputPtr);
Expand Down

0 comments on commit f2c2915

Please sign in to comment.