Replies: 3 comments
-
This is a different loop nest structure (the three outer loops have been fused), so you get at it using scheduling. We never change the loop structure specified by the schedule, so this optimization is not something Halide should do automatically. You want: sum.compute_at(conv, i) If there's no wrapper function to schedule it within, and it's called directly, you can use Func::in to make a wrapper: sum.compute_at(sum.in(), i); |
Beta Was this translation helpful? Give feedback.
-
It a new scheduling function, right? |
Beta Was this translation helpful? Give feedback.
-
Func::in is new. compute_at has existed a long time.
…On Mon, Dec 5, 2016 at 4:25 PM, Jian Weng ***@***.***> wrote:
It a new scheduling function, right?
I haven't updated my Halide for a long time.
Sorry.
—
You are receiving this because you commented.
Reply to this email directly, view it on GitHub
<https://github.com/halide/Halide/issues/1661#issuecomment-265022881>, or mute
the thread
<https://github.com/notifications/unsubscribe-auth/AAfdRr-VFVnkO5-riwXvJYAeoC6iUD9Xks5rFKuVgaJpZM4LEnof>
.
|
Beta Was this translation helpful? Give feedback.
-
Consider the simple convolution.
In the second stage of computation, the
Store
of arraysum
can be hoisted.Beta Was this translation helpful? Give feedback.
All reactions