File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -195,6 +195,8 @@ void getSegmentsOutputByRunning(
195
195
jit_inputs_ivalues.push_back (ivalues_maps[input].toBool ());
196
196
} else if (input->type ()->isSubtypeOf (torch::jit::FloatType::get ())) {
197
197
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 ());
198
200
} else if (input->type ()->kind () == torch::jit::TypeKind::ListType) {
199
201
// create list
200
202
jit_inputs_ivalues.push_back (ivalues_maps[input].toList ());
@@ -207,8 +209,6 @@ void getSegmentsOutputByRunning(
207
209
} else if (input->type ()->kind () == torch::jit::TypeKind::DictType) {
208
210
jit_inputs_ivalues.push_back (ivalues_maps[input].toGenericDict ());
209
211
} 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 ())) {
212
212
jit_inputs_ivalues.push_back (ivalues_maps[input].toString ());
213
213
} else {
214
214
TORCHTRT_THROW_ERROR (
You can’t perform that action at this time.
0 commit comments