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

CUTLASS 2.10 bug fixes and minor updates. (#626)

Browse Source
This commit is contained in:
Andrew Kerr
2022-09-15 16:20:33 -04:00
committed by GitHub
parent 2cc2c7ba1f
commit fc9ebc645b
5 changed files with 63 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/cutlass/epilogue/threadblock/predicated_tile_iterator.h
+2 -2
View File
@@ -671,7 +671,7 @@ public:
state_[1] = 0;
++state_[2];
byte_pointer_ += params_.advance_cluster;
store_byte_pointer_ += params_.advance_group;
store_byte_pointer_ += params_.advance_cluster;
thread_start_row_ += ThreadMap::Count::kGroup *
ThreadMap::Shape::kGroup * ThreadMap::Count::kRow * ThreadMap::Shape::kRow;
@@ -679,7 +679,7 @@ public:
if (state_[2] == ThreadMap::Count::kCluster) {
state_[2] = 0;
byte_pointer_ += params_.advance_tile;
store_byte_pointer_ += params_.advance_group;
store_byte_pointer_ += params_.advance_tile;
}
}
}