squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Increment stride correctly in warp iterator. (#516)

Browse Source 
Co-authored-by: peisun1115 <peis@google.com>
This commit is contained in:
Pei Sun 2022-06-06 09:33:36 -07:00 committed by GitHub
parent c3881d097e
commit dceefe4f64
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/cutlass/gemm/warp/mma_tensor_op_tile_iterator.h
View File

@ -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_;