Commit ad4a2be2 authored by MaYuhang's avatar MaYuhang
Browse files

issue/406: 修复 isMergable 的 stride 检查逻辑

parent cb06c721
......@@ -102,7 +102,7 @@ bool InfiniopTensorDescriptor::isMergable(size_t dim_start, size_t dim_end) cons
auto ndim_ = shape_.size();
for (size_t i = 1; i < ndim_; i++) {
if (stride(i - 1) != static_cast<ptrdiff_t>(dim(i)) * stride(i)) {
if (strides_[i - 1] != static_cast<ptrdiff_t>(shape_[i]) * strides_[i]) {
return false;
}
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment