Increment stride correctly in warp iterator. (#516)

Co-authored-by: peisun1115 <peis@google.com>
2022-06-06 12:33:36 -04:00
parent c3881d097e
commit dceefe4f64
1 changed files with 1 additions and 1 deletions
@@ -403,7 +403,7 @@ public:
        AccessType const *source_ptr =
            pointer_[c % kPointerCount] +
            Layout::TileShape::kContiguous * (c / kPointerCount) +
-            Policy::LdsmShape::kStrided * s * stride_;
+            Policy::kLdsmOpInner * Policy::LdsmShape::kStrided * s * stride_;

        char const *source_byte_ptr = reinterpret_cast<char const *>(source_ptr) + byte_offset + byte_offset_;