@@ -598,17 +598,15 @@ SetStringInputTensor(
598
598
}
599
599
#endif // TRITON_ENABLE_GPU
600
600
601
- auto callback = [](TRITONTF_Tensor* tensor, const size_t tensor_offset,
602
- const size_t element_idx, const char * content,
603
- const uint32_t len) {
604
- TRITONTF_TensorSetString (tensor, tensor_offset + element_idx, content, len);
605
- };
606
- auto fn = std::bind (
607
- callback, tensor, tensor_offset, std::placeholders::_1,
608
- std::placeholders::_2, std::placeholders::_3);
609
-
601
+ std::vector<std::pair<const char *, const uint32_t >> str_list;
610
602
err = ValidateStringBuffer (
611
- content, content_byte_size, request_element_cnt, name, &element_idx, fn);
603
+ content, content_byte_size, request_element_cnt, name, &str_list);
604
+ // Set string values.
605
+ for (; element_idx < str_list.size (); ++element_idx) {
606
+ const auto & [addr, len] = str_list[element_idx];
607
+ TRITONTF_TensorSetString (tensor, tensor_offset + element_idx, addr, len);
608
+ }
609
+
612
610
if (err != nullptr ) {
613
611
RESPOND_AND_SET_NULL_IF_ERROR (response, err);
614
612
FillStringTensor (
0 commit comments