You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.
MXNet supports 3D dilate on convolution. However, it doesn't leverage MKLDNN library.
As a result, one of our big customers faced the issue where GPU (3d conv dilate) ran in under 0.01 millisecond (as it leverages cuDNN library probably) while CPU takes ~200ms
Turns out MKLDNN since earlier versions v0.10 has been supporting conv dilate for 3d but with certain limitations. Maybe its time to port the functionality to MXNet and solve the customer problem.
MXNet supports 3D dilate on convolution. However, it doesn't leverage MKLDNN library.
As a result, one of our big customers faced the issue where GPU (3d conv dilate) ran in under 0.01 millisecond (as it leverages cuDNN library probably) while CPU takes ~200ms
Turns out MKLDNN since earlier versions v0.10 has been supporting conv dilate for 3d but with certain limitations. Maybe its time to port the functionality to MXNet and solve the customer problem.
Related issue
MKLDNN : oneapi-src/oneDNN#676
The text was updated successfully, but these errors were encountered: