From 182f6ad52e9fcab5deedc962bc5bf8d0118bafd2 Mon Sep 17 00:00:00 2001 From: Yingge He Date: Fri, 26 Jul 2024 11:02:23 -0700 Subject: [PATCH] Minor fix --- src/libtorch.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/libtorch.cc b/src/libtorch.cc index 7ce5c58..c4e964c 100644 --- a/src/libtorch.cc +++ b/src/libtorch.cc @@ -1911,7 +1911,6 @@ SetStringInputTensor( cudaStream_t stream, const char* host_policy_name) { bool cuda_copy = false; - size_t element_idx = 0; // For string data type, we always need to have the data on CPU so // that we can read string length and construct the string @@ -1926,7 +1925,7 @@ SetStringInputTensor( stream, &cuda_copy); if (err != nullptr) { RESPOND_AND_SET_NULL_IF_ERROR(response, err); - FillStringTensor(input_list, request_element_cnt - element_idx); + FillStringTensor(input_list, request_element_cnt); return cuda_copy; } @@ -1945,9 +1944,10 @@ SetStringInputTensor( input_list->push_back(std::string(addr, len)); } + size_t element_cnt = str_list.size(); if (err != nullptr) { RESPOND_AND_SET_NULL_IF_ERROR(response, err); - FillStringTensor(input_list, request_element_cnt - element_idx); + FillStringTensor(input_list, request_element_cnt - element_cnt); } return cuda_copy; }