[OPTIMIZER] Rewrite patterns for layout conversions (#64)

2022-08-18 12:49:37 -07:00
parent e0bedeb44c
commit 192be76b3c
19 changed files with 851 additions and 127 deletions
--- a/lib/Analysis/Allocation.cpp
+++ b/lib/Analysis/Allocation.cpp
@@ -43,8 +43,7 @@ private:
      auto type = result.getType();
      if (auto tensorType = type.dyn_cast<RankedTensorType>()) {
        auto encoding = tensorType.getEncoding();
-        if (encoding &&
-            encoding.isa<triton::gpu::TritonGPUSharedEncodingAttr>()) {
+        if (encoding && encoding.isa<triton::gpu::SharedEncodingAttr>()) {
          // Bytes could be a different value once we support padding or other
          // allocation policies.
          auto bytes = tensorType.getNumElements() *