fix: include str ivalue type conversion

Author: bowang007

core/partitioning/shape_analysis.cpp

@@ -195,6 +195,8 @@ void getSegmentsOutputByRunning(
       jit_inputs_ivalues.push_back(ivalues_maps[input].toBool());
     } else if (input->type()->isSubtypeOf(torch::jit::FloatType::get())) {
       jit_inputs_ivalues.push_back(ivalues_maps[input].toDouble());
+    } else if (input->type()->isSubtypeOf(torch::jit::StringType::get())) {
+      jit_inputs_ivalues.push_back(ivalues_maps[input].toString());
     } else if (input->type()->kind() == torch::jit::TypeKind::ListType) {
       // create list
       jit_inputs_ivalues.push_back(ivalues_maps[input].toList());
@@ -207,7 +209,7 @@ void getSegmentsOutputByRunning(
     } else if (input->type()->kind() == torch::jit::TypeKind::DictType) {
       jit_inputs_ivalues.push_back(ivalues_maps[input].toGenericDict());
     } else if (input->type()->kind() == torch::jit::TypeKind::DeviceObjType) {
-      jit_inputs_ivalues.push_back(ivalues_maps[input].toDevice());
+      jit_inputs_ivalues.push_back(ivalues_maps[input].toString());
     } else {
       TORCHTRT_THROW_ERROR(
           "Expected to find type " << input->type()->str() << " for value " << input->debugName()