From f963d53415c7a0cf91809a806ea83a30eabd2ec1 Mon Sep 17 00:00:00 2001 From: HongHongHongL Date: Thu, 21 Nov 2024 19:11:06 +0800 Subject: [PATCH] fix segment fault when get undef layout --- src/relax/transform/convert_layout.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/relax/transform/convert_layout.cc b/src/relax/transform/convert_layout.cc index 2048f0ddedf5..3798bdff351d 100644 --- a/src/relax/transform/convert_layout.cc +++ b/src/relax/transform/convert_layout.cc @@ -90,7 +90,7 @@ class LayoutConvertMutator : public ExprMutator { Expr RewriteExpr(const Expr& expr, const NLayout& to) { auto fvisitleaf = [&](const Expr& expr, std::array layouts) -> Expr { NLayout from = layouts[0], to = layouts[1]; - if (NLayoutEqual()(from, to) || layouts[0].LeafValue()->layout->name == "") return expr; + if (NLayoutEqual()(from, to) || layouts[0].LeafValue()->layout.name() == "") return expr; // If not both from and to are unknown, then none of them can be unknown. ICHECK(!NLayoutEqual()(from, LayoutDecision::InitUnknownDim()) && !NLayoutEqual()(to, LayoutDecision::InitUnknownDim()))