We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ef62f6b commit 01c89d1Copy full SHA for 01c89d1
core/compiler.cpp
@@ -329,6 +329,7 @@ void MapInputsAndDetermineDTypes(
329
} else if (spec.dtype_is_user_defined && cfg.partition_info.enabled) {
330
if (!est_type_opt) {
331
LOG_INFO("Cannot infer input tensor dtype in graph, unable to verify user input dtype settings");
332
+ first_use_type_map[in] = {util::TRTDataTypeToScalarType(cfg.convert_info.inputs.find(in)->second.dtype)};
333
} else {
334
if (util::TRTDataTypeToScalarType(cfg.convert_info.inputs.find(in)->second.dtype) != est_type_opt.value()) {
335
std::stringstream ss;
0 commit comments