Skip to content

Commit

Permalink
fix(//core): Take user setting in the case we can't determine the
Browse files Browse the repository at this point in the history
inferred type.

fixes: #814

Signed-off-by: Naren Dasan <naren@narendasan.com>
Signed-off-by: Naren Dasan <narens@nvidia.com>
  • Loading branch information
narendasan committed Mar 1, 2022
1 parent ef62f6b commit 01c89d1
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions core/compiler.cpp
Expand Up @@ -329,6 +329,7 @@ void MapInputsAndDetermineDTypes(
} else if (spec.dtype_is_user_defined && cfg.partition_info.enabled) {
if (!est_type_opt) {
LOG_INFO("Cannot infer input tensor dtype in graph, unable to verify user input dtype settings");
first_use_type_map[in] = {util::TRTDataTypeToScalarType(cfg.convert_info.inputs.find(in)->second.dtype)};
} else {
if (util::TRTDataTypeToScalarType(cfg.convert_info.inputs.find(in)->second.dtype) != est_type_opt.value()) {
std::stringstream ss;
Expand Down

0 comments on commit 01c89d1

Please sign in to comment.