Skip to content

Commit 62a60d6

Browse files
authored
Compare full key value when retrieving named data
Differential Revision: D82129824 Pull Request resolved: #14165
1 parent f7df6b1 commit 62a60d6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

extension/flat_tensor/flat_tensor_data_map.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,8 @@ Result<const flat_tensor_flatbuffer::NamedData*> get_named_data(
5656
return Error::NotFound;
5757
}
5858
for (int i = 0; i < named_data->size(); i++) {
59-
if (std::strncmp(
59+
if (key.size() == named_data->Get(i)->key()->size() &&
60+
std::strncmp(
6061
named_data->Get(i)->key()->c_str(),
6162
key.data(),
6263
named_data->Get(i)->key()->size()) == 0) {

0 commit comments

Comments
 (0)