-
Notifications
You must be signed in to change notification settings - Fork 1.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[adams2019] Restructure autoscheduler + add timer #5654
Conversation
…lex/add_autosched_caching
Buildbots are green (ignoring the buildbot-beta); looks ready to be reviewed. |
This is large and hard to review. It looks like there are two things here - a refactor of the autoscheduler into more files, and adding caching. Is it possible to disentangle them into separate PRs (e.g. by deleting everything caching related from this PR, and then reintroducing it in another?). I can't tell what's moved code and what's new. |
@abadams This PR should now just be restructuring (+ the addition of the timer, but I can remove that and put it in the caching PR if necessary). |
lgtm with one final suggested change. |
Thanks, you are good to merge once the bots are all green. |
Failures look unrelated to the autoscheduler, @steven-johnson can you confirm? |
Failures are unrelated, OK to land |
Adds caching of features and schedule enumerations to the adams2019 autoscheduler. Backported from @aekul 's autoscheduler work.On my machine, I see a x2 speedup on the time needed to autoschedule lens blur and local laplacian, about a x1.5 speedup for resnet50, and anywhere between x1-1.5 for other pipelines (caching is only useful for larger pipelines, it has little to no effect on smaller pipelines).This PR is now just a restructuring + addition of a timer to the adams2019 autoscheduler, to allow for the addition of caching.