Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
[1.x][submodule] Upgrade oneDNN to the top of rls-v2.4 branch (#20989)
Browse files Browse the repository at this point in the history
  • Loading branch information
bartekkuncer authored Apr 8, 2022
1 parent 4d399c2 commit 40c7e09
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 8 deletions.
8 changes: 1 addition & 7 deletions src/operator/nn/mkldnn/mkldnn_convolution.cc
Original file line number Diff line number Diff line change
Expand Up @@ -123,13 +123,7 @@ std::shared_ptr<mkldnn::convolution_forward::primitive_desc> GetConvFwdImpl(
conv_pd->dst_desc().get_size() != GetArraySize(output) ||
conv_pd->src_desc().get_size() != GetArraySize(data) ||
(!param.mkldnn_param.quantized &&
conv_pd->weights_desc().get_size() != GetArraySize(weights)) ||
// With the upgrade of MKLDNN to version 2.4+
// tests/python/mkl/test_subgraph.py::test_pos_conv_add started failing. Switching away
// from blocking weights in order to temporarily fix the issue until full fix arrives.
// Tracking issue: https://github.com/apache/incubator-mxnet/issues/20826.
(param.mkldnn_param.quantized && conv_pd->weights_desc().dims()[1] < 4 &&
conv_pd->weights_desc().data.padded_dims[1] != conv_pd->weights_desc().dims()[1])) {
conv_pd->weights_desc().get_size() != GetArraySize(weights))) {
// next_impl() will visit desc and engine, please make sure they are still alive here.
CHECK(conv_pd->next_impl()) << "No convolution implementation for this request.";
}
Expand Down

0 comments on commit 40c7e09

Please sign in to comment.