We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ad5e764 commit 4e61dfaCopy full SHA for 4e61dfa
core/partitioning/shape_analysis.cpp
@@ -208,6 +208,8 @@ void getSegmentsOutputByRunning(
208
jit_inputs_ivalues.push_back(ivalues_maps[input].toGenericDict());
209
} 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
+ jit_inputs_ivalues.push_back(ivalues_maps[input].toString());
213
} else {
214
TORCHTRT_THROW_ERROR(
215
"Expected to find type " << input->type()->str() << " for value " << input->debugName()
0 commit comments