Skip to content

Commit 0591a8d

Browse files
committed
fix: fix misleading skipping partitioning msg
Signed-off-by: Bo Wang <[email protected]>
1 parent 9bb0087 commit 0591a8d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/compiler.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -426,7 +426,7 @@ torch::jit::Module CompileGraph(const torch::jit::Module& mod, CompileSpec cfg)
426426
auto outputIsCollection = conversion::OutputIsCollection(g->block());
427427
if (cfg.partition_info.enabled &&
428428
(cfg.lower_info.forced_fallback_modules.size() == 0 &&
429-
cfg.partition_info.forced_fallback_operators.size() == 0 && isBlockConvertible)) {
429+
cfg.partition_info.forced_fallback_operators.size() == 0 && isBlockConvertible) && !outputIsCollection) {
430430
LOG_INFO("Skipping partitioning since model is fully supported");
431431
}
432432

0 commit comments

Comments
 (0)