[mlir][bufferize] Update remaining getMemorySpaceAsInt API uses

[springerm]

Summary

D138330 updated the deprecated getMemorySpaceAsInt uses to getMemorySpace. There are few uses that were missed.

mlir/lib/Dialect/Bufferization/IR/BufferizationOps.cpp

 FailureOr<Value>
 mlir::bufferization::castOrReallocMemRefValue(OpBuilder &b, Value value,
                                               MemRefType destType) {
   auto srcType = value.getType().cast<MemRefType>();
 
   // Element type, rank and memory space must match.
   if (srcType.getElementType() != destType.getElementType())
     return failure();
-  if (srcType.getMemorySpaceAsInt() != destType.getMemorySpaceAsInt())
+  if (srcType.getMemorySpace() != destType.getMemorySpace())
     return failure();
   if (srcType.getRank() != destType.getRank())
     return failure();
 
   // In case the affine maps are different, we may need to use a copy if we go
   // from dynamic to static offset or stride (the canonicalization cannot know
   // at this point that it is really cast compatible).
   auto isGuaranteedCastCompatible = [](MemRefType source, MemRefType target) {

mlir/lib/Dialect/Bufferization/Transforms/BufferResultsToOutParams.cpp

for (Value memref : replaceWithOutParams) {
         op.emitError()
             << "cannot create out param for dynamically shaped result";
         didFail = true;
         return;
       }
       auto memrefType = memref.getType().cast<MemRefType>();
       auto allocType =
           MemRefType::get(memrefType.getShape(), memrefType.getElementType(),
-                          AffineMap(), memrefType.getMemorySpaceAsInt());
+                          AffineMap(), memrefType.getMemorySpace());
       Value outParam = builder.create<memref::AllocOp>(op.getLoc(), allocType);
       if (!hasStaticIdentityLayout(memrefType)) {
         // Layout maps are already checked in `updateFuncOp`.
         assert(hasFullyDynamicLayoutMap(memrefType) &&
                "layout map not supported");
         outParam =
             builder.create<memref::CastOp>(op.getLoc(), memrefType, outParam);
       }