Skip to content

Commit 4ac19be

Browse files
maxnickluweizhou2016
authored andcommitted
[FIX] Desc similar_to routine consider start stride
1 parent c8eb642 commit 4ac19be

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
@@ -602,8 +602,11 @@ inline bool memory_desc_wrapper::similar_to(const memory_desc_wrapper &rhs,
602602
if (is_wino_desc() || is_rnn_packed_desc()) return false;
603603

604604
const int ds = dim_start;
605-
if (stride_start == -1)
605+
if (stride_start == -1) {
606606
stride_start = ds;
607+
} else if (stride_start > ndims()) {
608+
stride_start = ndims();
609+
}
607610
const auto &blk = blocking_desc();
608611
const auto &r_blk = rhs.blocking_desc();
609612

@@ -612,7 +615,7 @@ inline bool memory_desc_wrapper::similar_to(const memory_desc_wrapper &rhs,
612615
&& format_kind() == rhs.format_kind()
613616
&& IMPLICATION(with_data_type, data_type() == rhs.data_type())
614617
&& custom_cpm(dims() + ds, rhs.dims() + ds, ndims() - ds)
615-
&& custom_cpm(blk.strides + ds, r_blk.strides + ds, ndims() - ds)
618+
&& custom_cpm(blk.strides + stride_start, r_blk.strides + stride_start, ndims() - stride_start)
616619
&& blk.inner_nblks == r_blk.inner_nblks
617620
&& array_cmp(blk.inner_blks, r_blk.inner_blks, blk.inner_nblks)
618621
&& array_cmp(blk.inner_idxs, r_blk.inner_idxs, blk.inner_nblks)

0 commit comments

Comments
 (0)