search

cornell-zhang / heterocl

HeteroCL: A Multi-Paradigm Programming Infrastructure for Software-Defined Heterogeneous Computing
https://cornell-zhang.github.io/heterocl/
Apache License 2.0
322 stars 92 forks source link

Enable applying multiple axes in a single primitive #126

Open seanlatias opened 5 years ago

seanlatias commented 5 years ago

We need to add the support for this. Something like s[A].unroll([A.axis[0], A.axis[1])