CUTLASS 2.5

include/cutlass/epilogue/threadblock/shared_load_iterator.h

@@ -1,5 +1,5 @@
 /***************************************************************************************************
- * Copyright (c) 2017-2020, NVIDIA CORPORATION.  All rights reserved.
+ * Copyright (c) 2017-2021, NVIDIA CORPORATION.  All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without modification, are permitted
  * provided that the following conditions are met:
@@ -61,7 +61,7 @@ template <
 class SharedLoadIterator {
 public:
   using ThreadMap = ThreadMap_;
-  using Shape = typename ThreadMap::Shape;
+  using Shape = typename ThreadMap::TileShape;
 
   using Element = Element_;
 
@@ -151,7 +151,9 @@ public:
 
   CUTLASS_DEVICE
   void add_tile_offset(TensorCoord const &offset) {
-    add_pointer_offset(offset.row() * stride_ / (sizeof_bits<Element>::value / 8) + offset.column() * Shape::kColumn);
+    byte_pointer_ += 
+      offset.row() * Shape::kRow * stride_ + 
+      offset.column() * Shape::kColumn * sizeof_bits<Element>::value / 8;
   }
 
   /// Loads a fragment from memory