CINN
CINN copied to clipboard
Add Conv2d_winograd, fix `Fuse` schedule and enhance simplify
This PR did 3 things:
- Add conv2d_winograd compute and schedule
- Fix Fuse schedule
- Enhance simplify: For examples: a = 2x and b = 8 -> a/b = x/4
if 0 < b < 3 -> (3a+b) % 6 = (3a % 6) + (b % 6)
2x % 4 = 2 * (x % 2)
if 0 < b < 3, (3a+b) / 6 = (3a / 6) + (b / 6)
Thanks for your contribution!