search

cornell-zhang / hcl-dialect

HeteroCL-MLIR dialect for accelerator design
https://cornell-zhang.github.io/heterocl/index.html
Other
40 stars 17 forks source link

[Frontend] Add axis to the update stage #111

Closed zzzDavid closed 1 year ago

zzzDavid commented 2 years ago

Related test: mlir/test_dsl_def.py::test_module_declarative_compute_at

        @hcl.def_([a.shape, b.shape, c.shape])
        def add(a, b, c):
            d = hcl.compute(a.shape, lambda *x: a[x] + b[x], "d")
            hcl.update(c, lambda *x: d[x] + 1, "u") # this stage `u` has an empty axis list
zzzDavid commented 1 year ago

Fixed by cornell-zhang/heterocl@47516b0353c4988572c73514575bb12c6b4449ca