Skip to content

Commit 4e61dfa

Browse files
committed
fix: include str ivalue type conversion
1 parent ad5e764 commit 4e61dfa

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

core/partitioning/shape_analysis.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,8 @@ void getSegmentsOutputByRunning(
208208
jit_inputs_ivalues.push_back(ivalues_maps[input].toGenericDict());
209209
} else if (input->type()->kind() == torch::jit::TypeKind::DeviceObjType) {
210210
jit_inputs_ivalues.push_back(ivalues_maps[input].toDevice());
211+
} else if (input->type() == (torch::jit::StringType::get())) {
212+
jit_inputs_ivalues.push_back(ivalues_maps[input].toString());
211213
} else {
212214
TORCHTRT_THROW_ERROR(
213215
"Expected to find type " << input->type()->str() << " for value " << input->debugName()

0 commit comments

Comments
 (0)