Skip to content

Commit 0772af8

Browse files
committed
chore: reformat
1 parent 4e61dfa commit 0772af8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

core/partitioning/shape_analysis.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -195,6 +195,8 @@ void getSegmentsOutputByRunning(
195195
jit_inputs_ivalues.push_back(ivalues_maps[input].toBool());
196196
} else if (input->type()->isSubtypeOf(torch::jit::FloatType::get())) {
197197
jit_inputs_ivalues.push_back(ivalues_maps[input].toDouble());
198+
} else if (input->type()->isSubtypeOf(torch::jit::StringType::get())) {
199+
jit_inputs_ivalues.push_back(ivalues_maps[input].toString());
198200
} else if (input->type()->kind() == torch::jit::TypeKind::ListType) {
199201
// create list
200202
jit_inputs_ivalues.push_back(ivalues_maps[input].toList());
@@ -207,8 +209,6 @@ void getSegmentsOutputByRunning(
207209
} else if (input->type()->kind() == torch::jit::TypeKind::DictType) {
208210
jit_inputs_ivalues.push_back(ivalues_maps[input].toGenericDict());
209211
} else if (input->type()->kind() == torch::jit::TypeKind::DeviceObjType) {
210-
jit_inputs_ivalues.push_back(ivalues_maps[input].toDevice());
211-
} else if (input->type() == (torch::jit::StringType::get())) {
212212
jit_inputs_ivalues.push_back(ivalues_maps[input].toString());
213213
} else {
214214
TORCHTRT_THROW_ERROR(

0 commit comments

Comments
 (0)