TTIR Bufferization pass (#2264)

### What's changed
This change implements a bufferization interface for the ttir generic op
and to_layout op and invokes the relevant passes. This also means that
these ops can now accept memrefs as inputs.

Some excellent documentation on the bufferization pass can be found here
https://mlir.llvm.org/docs/Bufferization/. The motivation in our case is
to leverage this upstream machinery and get "for free":
- Automatic analysis of buffer allocation, aliasing, deallocation
- Type conversion from tensors to buffer objects
- Seamless integration with other dialects used at this stage, memref,
linalg, affine, etc.

### Checklist
- [x] New/Existing tests provide coverage for changes

include/ttmlir/Dialect/TT/IR/TTOpsTypes.td

@@ -351,6 +351,7 @@ def TT_MetalLayoutAttr : TT_Attr<"MetalLayout", "metal_layout"> {
       MetalLayoutAttr withMemorySpace(::mlir::MLIRContext *context, MemorySpace memorySpace);
       MetalLayoutAttr withShardShape(::mlir::MLIRContext *context, llvm::SmallVector<int64_t> shardShape);
       MetalLayoutAttr withStreamLayout(::mlir::MLIRContext *context, StreamLayoutAttr stream);
+      MetalLayoutAttr withStreamMode(::mlir::MLIRContext *context, StreamMode streamMode, std::uint32_t numBuffers);
       MetalLayoutAttr withOuterScale(::mlir::MLIRContext *context,
                           llvm::ArrayRef<int64_t> outerScale,
                           StreamMode streamMode,

include/ttmlir/Dialect/TTIR/IR/TTIRBase.td

@@ -42,7 +42,7 @@ def TTIR_Dialect : Dialect {
 //===----------------------------------------------------------------------===//
 
 class TTIR_Op<string mnemonic, list<Trait> traits = []> :
-        Op<TTIR_Dialect, mnemonic, !listconcat([Pure], traits)>;
+        Op<TTIR_Dialect, mnemonic, traits>;
 
 //===----------------------------------------------------------------------===//
 // TTIR traits definition.
@@ -68,4 +68,10 @@ def TTIR_BinaryIdempotence : TTIR_Trait<"TTIRBinaryIdempotence", [DestinationSty
 // GenericRegionOpTrait is a trait that acts as a label for all generic region operations.
 def TTIR_GenericRegionOpTrait : TTIR_Trait<"TTIRGenericRegionOpTrait", []>;
 
+//===----------------------------------------------------------------------===//
+// TTIR common types.
+//===----------------------------------------------------------------------===//
+
+def AnyRankedTensorOrMemRef: AnyTypeOf<[AnyRankedTensor, AnyNon0RankedMemRef]>;
+
 #endif

include/ttmlir/Dialect/TTIR/IR/TTIRGenericRegionOps.td

@@ -20,8 +20,6 @@ class TTIR_GenericRegionOp<string mnemonic, list<Trait> traits = [TTIR_GenericRe
 // TTIR Generic Region Ops (Used in TTMetal Lowering)
 //===----------------------------------------------------------------------===//
 
-def AnyRankedTensorOrMemRef: AnyTypeOf<[AnyRankedTensor, AnyNon0RankedMemRef]>;
-
 def TTIR_YieldOp : TTIR_Op<"yield", [Pure, ReturnLike, Terminator, TTIR_GenericRegionOpTrait]> {
     let summary = "Yield op.";
     let description = [{

include/ttmlir/Dialect/TTIR/IR/TTIROps.h

@@ -10,6 +10,7 @@
 #include "ttmlir/Dialect/TTIR/IR/TTIRTraits.h"
 
 #include "mlir/Bytecode/BytecodeOpInterface.h"
+#include "mlir/Dialect/Bufferization/IR/BufferizableOpInterface.h"
 #include "mlir/IR/BuiltinTypes.h"
 #include "mlir/IR/OpDefinition.h"
 #include "mlir/IR/PatternMatch.h"