We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents e20d70a + 37b5d5c commit a245b86Copy full SHA for a245b86
core/partitioning/shape_analysis.cpp
@@ -195,6 +195,8 @@ void getSegmentsOutputByRunning(
195
jit_inputs_ivalues.push_back(ivalues_maps[input].toBool());
196
} else if (input->type()->isSubtypeOf(torch::jit::FloatType::get())) {
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());
200
} else if (input->type()->kind() == torch::jit::TypeKind::ListType) {
201
// create list
202
jit_inputs_ivalues.push_back(ivalues_maps[input].toList());
0 commit comments