From 0591a8d137473417f57e27402bbbfd142d5ef397 Mon Sep 17 00:00:00 2001 From: Bo Wang Date: Thu, 18 Aug 2022 14:49:14 -0700 Subject: [PATCH] fix: fix misleading skipping partitioning msg Signed-off-by: Bo Wang --- core/compiler.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/compiler.cpp b/core/compiler.cpp index 7b58dbb2c1..898f2f1295 100644 --- a/core/compiler.cpp +++ b/core/compiler.cpp @@ -426,7 +426,7 @@ torch::jit::Module CompileGraph(const torch::jit::Module& mod, CompileSpec cfg) auto outputIsCollection = conversion::OutputIsCollection(g->block()); if (cfg.partition_info.enabled && (cfg.lower_info.forced_fallback_modules.size() == 0 && - cfg.partition_info.forced_fallback_operators.size() == 0 && isBlockConvertible)) { + cfg.partition_info.forced_fallback_operators.size() == 0 && isBlockConvertible) && !outputIsCollection) { LOG_INFO("Skipping partitioning since model is fully supported"); }