-
Notifications
You must be signed in to change notification settings - Fork 3.8k
[AutoTVM][TOPI] AutoTVM support for NHWC con2d #3858
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
Conversation
|
Currently WIP since schedule templates will need to be fine-tuned to reach good performance on AVX |
|
@tmoreau89 Very glad to see this :) I have a couple of high-level comments/questions
|
|
@anijain2305 thanks for the feedback; I'll need to fine tune the schedules and report on the perf numbers, hence the WIP label! Also is there potential for code reuse between x86 and ARM schedules? |
|
Good question about what is deprecated. I don't know actually. I will try to figure that out. |
|
My feeling is that if x86 NHWC doesn't show performance improvement over NCHWc, probably we should go this way: #3670. |
4a5673d to
66742f7
Compare
66742f7 to
2815c30
Compare
|
ping @tmoreau89 please look into the CI error |
|
@tqchen thanks for the ping, this PR has been a little dormant and requires some work, I'll place a WIP tag on it. The main issue right now is to get a schedule template that matches NCHWc in terms of performance. |
|
superseded by #3859 |
This PR brings NHWC conv2d TOPI declaration and schedules for x86 (ARM support to follow)
This addresses the lack of autotuning support pointed out in the discussion in #3141