Uh oh!
There was an error while loading. Please reload this page.
[AutoTVM][TOPI] AutoTVM support for NHWC con2d - #3858
Conversation
tmoreau89
commented
Aug 30, 2019
Currently WIP since schedule templates will need to be fine-tuned to reach good performance on AVX |
anijain2305
commented
Aug 30, 2019
@tmoreau89 Very glad to see this :) I have a couple of high-level comments/questions
|
tmoreau89
commented
Aug 30, 2019
@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? |
anijain2305
commented
Aug 30, 2019
Good question about what is deprecated. I don't know actually. I will try to figure that out. |
kevinthesun
commented
Aug 30, 2019
My feeling is that if x86 NHWC doesn't show performance improvement over NCHWc, probably we should go this way: #3670. |
tqchen
commented
Sep 13, 2019
ping @tmoreau89 please look into the CI error |
tmoreau89
commented
Sep 14, 2019
@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. |
tqchen
commented
Jan 15, 2020
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