This repository has been archived by the owner on Nov 17, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Conv1D is slow #11161
Labels
Comments
Yes. By doing this, conv1d can also benefit from mkldnn conv2d on cpu side. @pengzhao-intel |
In our local case, it can be 10x faster :) We can provide the PR for CPU side @jinhuang415 |
That will be great! |
@pengzhao-intel @jinhuang415 is there a PR for this issue? |
@vandanavk Not yet, the 1D conv is still under the development. |
@xinyu-intel please check the latest MKL-DNN and conv1d. |
7 tasks
The performance is listed in this PR @eric-haibin-lin |
Sign up for free
to subscribe to this conversation on GitHub.
Already have an account?
Sign in.
The Conv1D block uses MXNet's implementation for Convolution, because CUDNN only implements 2-D convolution. In the operator, we can try to optimize the performance by reshaping the inputs to 4D, and kernels to 2D, so that CUDNN kernel can be used.
The text was updated successfully, but these errors were encountered: