From 267d612723a73f17e0816338693e24aaef07e798 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 12 Jan 2021 16:49:15 +0800 Subject: [PATCH] fix(imperative/mini_graph): fix no_err value in shape infer GitOrigin-RevId: 087d15db661b2ea366a007eccfd15bd94ae58d79 --- imperative/src/impl/proxy_graph/mini_graph.h | 1 + 1 file changed, 1 insertion(+) diff --git a/imperative/src/impl/proxy_graph/mini_graph.h b/imperative/src/impl/proxy_graph/mini_graph.h index a3866b4f5..52aa55679 100644 --- a/imperative/src/impl/proxy_graph/mini_graph.h +++ b/imperative/src/impl/proxy_graph/mini_graph.h @@ -606,6 +606,7 @@ public: desc.comp_node = minigraph.output_var(i)->comp_node(); if (auto* shape = sess.infer_shape(i, false)) { desc.layout.init_contiguous_stride(*shape); + noerr = true; } else { noerr = false; } -- GitLab