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(
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 (
You can’t perform that action at this time.
0 commit comments