Skip to content

Commit 01c89d1

Browse files
committed
fix(//core): Take user setting in the case we can't determine the
inferred type. fixes: #814 Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent ef62f6b commit 01c89d1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Diff for: core/compiler.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -329,6 +329,7 @@ void MapInputsAndDetermineDTypes(
329329
} else if (spec.dtype_is_user_defined && cfg.partition_info.enabled) {
330330
if (!est_type_opt) {
331331
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)};
332333
} else {
333334
if (util::TRTDataTypeToScalarType(cfg.convert_info.inputs.find(in)->second.dtype) != est_type_opt.value()) {
334335
std::stringstream ss;

0 commit comments

Comments
 (0)