search

apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators
https://tvm.apache.org/
Apache License 2.0
11.41k stars 3.4k forks source link

[Bugfix] Update FAttrsGetter to return Map<String, ObjectRef> #17096

Closed Lunderberg closed 2 weeks ago

Lunderberg commented 3 weeks ago

Prior to this commit, FAttrsGetter was defined as a function that returned Map<String, String>. However, it is used to define attributes in a Map<String, ObjectRef>, 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<String, ObjectRef>.

Lunderberg commented 3 weeks ago

This change is split out from https://github.com/apache/tvm/pull/16183 as an independent change for ease of review.