From 7a41185a2d83127ad966b11c6bf2a7be22b8b40c Mon Sep 17 00:00:00 2001 From: Eric Lunderberg Date: Fri, 14 Jun 2024 15:06:34 -0500 Subject: [PATCH] [Bugfix] Update FAttrsGetter to return Map Prior to this commit, `FAttrsGetter` was defined as a function that returned `Map`. However, it is used to define attributes in a `Map`, and in some cases is used to define attributes whose value is a dictionary (e.g. `msc_attrs_getter` in `python/tvm/contrib/msc/core/transform/pattern.py`). This commit updates the type signature of `FAttrsGetter` to match its usage, returning a `Map`. --- include/tvm/relax/transform.h | 2 +- src/relax/transform/fuse_ops.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/include/tvm/relax/transform.h b/include/tvm/relax/transform.h index c3a3c873c02b..d8f36e478669 100644 --- a/include/tvm/relax/transform.h +++ b/include/tvm/relax/transform.h @@ -367,7 +367,7 @@ class FusionPatternNode : public Object { * \brief The function to get attributes for fused function * * It should have signature - * Map(const Map& context) + * Map(const Map& context) */ Optional attrs_getter; diff --git a/src/relax/transform/fuse_ops.cc b/src/relax/transform/fuse_ops.cc index c4bd52eff18e..45d70fc3e290 100644 --- a/src/relax/transform/fuse_ops.cc +++ b/src/relax/transform/fuse_ops.cc @@ -1035,7 +1035,7 @@ class PatternBasedPartitioner : ExprVisitor { using PatternCheckContext = transform::PatternCheckContext; using ExprVisitor::VisitExpr_; using FCheckMatch = runtime::TypedPackedFunc; - using FAttrsGetter = runtime::TypedPackedFunc(const Map&)>; + using FAttrsGetter = runtime::TypedPackedFunc(const Map&)>; static GroupMap Run(String pattern_name, DFPattern pattern, Map annotation_patterns, FCheckMatch check, Expr expr,