Skip to content

Commit 6fc4984

Browse files
committed
fix(autoware_tensorrt_plugins): use mutable unique temp storage size
1 parent 6b8a72d commit 6fc4984

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • perception/autoware_tensorrt_plugins/src/unique_ops

perception/autoware_tensorrt_plugins/src/unique_ops/unique.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ cudaError_t unique(
198198
return cudaMemsetAsync(num_unique, 0, sizeof(std::int64_t), stream);
199199
}
200200

201-
const auto temp_storage_size = get_unique_temp_storage_size(num_input_elements);
201+
std::size_t temp_storage_size = get_unique_temp_storage_size(num_input_elements);
202202
const auto scratch_offset = align_up(temp_storage_size, alignof(std::int64_t));
203203
auto * scratch = reinterpret_cast<char *>(workspace) + scratch_offset;
204204

0 commit comments

Comments
 (0)