search

tlc-pack / relax

Apache License 2.0
193 stars 58 forks source link

[Bugfix] [Pass] Fix AnnotateTIROpPattern fail on cumsum #369

Closed psrivas2 closed 1 year ago

psrivas2 commented 1 year ago

This PR changes the behavior of AnnotateTIROpPattern pass to classify PrimFuncs that have multiple buffer stores in a block as Opaque, instead of assert fail. This pattern is seen in cumsum operations.

cc: @Hzfengsy @YuchenJin

psrivas2 commented 1 year ago

Thanks @Hzfengsy for the review.