Skip to content

Commit 0d3afe5

Browse files
maxnickluweizhou2016
authored andcommitted
[FIX] Desc similar_to routine consider start stride
1 parent f9a5da3 commit 0d3afe5

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/common/memory_desc_wrapper.hpp

+5-2
Original file line numberDiff line numberDiff line change
@@ -597,8 +597,11 @@ inline bool memory_desc_wrapper::similar_to(const memory_desc_wrapper &rhs,
597597
if (is_wino_desc() || is_rnn_packed_desc()) return false;
598598

599599
const int ds = dim_start;
600-
if (stride_start == -1)
600+
if (stride_start == -1) {
601601
stride_start = ds;
602+
} else if (stride_start > ndims()) {
603+
stride_start = ndims();
604+
}
602605
const auto &blk = blocking_desc();
603606
const auto &r_blk = rhs.blocking_desc();
604607

@@ -607,7 +610,7 @@ inline bool memory_desc_wrapper::similar_to(const memory_desc_wrapper &rhs,
607610
&& format_kind() == rhs.format_kind()
608611
&& IMPLICATION(with_data_type, data_type() == rhs.data_type())
609612
&& custom_cpm(dims() + ds, rhs.dims() + ds, ndims() - ds)
610-
&& custom_cpm(blk.strides + ds, r_blk.strides + ds, ndims() - ds)
613+
&& custom_cpm(blk.strides + stride_start, r_blk.strides + stride_start, ndims() - stride_start)
611614
&& blk.inner_nblks == r_blk.inner_nblks
612615
&& array_cmp(blk.inner_blks, r_blk.inner_blks, blk.inner_nblks)
613616
&& array_cmp(blk.inner_idxs, r_blk.inner_idxs, blk.inner_nblks)

0 commit comments

Comments
 (0)