Fix FreeOp shape.

FreeOp type parsing and printing was inconsistent. FreeOp shape was invalid in cases where the operand mapping was not the identity. Change FreeOp so that:
* shape() returns `mapping(operand_shape)`, previously was `inverse_mapping(operand_shape)`
* the printed type is `mapping(operand_type)`, previously was `operand_type`
* parsed operand type remains `inverse_mapping(printed_type)`.

PiperOrigin-RevId: 361498431

Author: ulysseB

sair_ops.cc

@@ -781,7 +781,8 @@ static void Print(SairFreeOp op, mlir::OpAsmPrinter &printer) {
   PrintValueAccess(op.Value(), printer);
   printer.printOptionalAttrDict(op->getAttrs(),
                                 {SairDialect::kMappingAttrName});
-  printer << " : " << op.value().getType();
+  mlir::Type element_type = op.Value().GetType().ElementType();
+  printer << " : " << ValueType::get(op.shape(), element_type);
 }
 
 mlir::LogicalResult Verify(SairFromScalarOp op) {
@@ -1648,8 +1649,8 @@ llvm::SmallVector<int, 2> SairFreeOp::SubDomains() {
 }
 
 DomainShapeAttr SairFreeOp::shape() {
-  return value().getType().cast<ValueType>().Shape().AccessedShape(
-      mapping_array()[0].cast<MappingAttr>());
+  ValueOperand value = Value();
+  return value.GetType().Shape().AccessedShape(value.Mapping().Inverse());
 }
 
 // Takes a mapping `lhs` and an array of mappings `rhs_array`. Returns a new

test/roundtrip.mlir

@@ -656,3 +656,17 @@ func @placeholder_with_loop_nest(%arg0: f32) {
   }
   return
 }
+
+// CHECK-LABEL: @free_with_mapping
+func @free_with_mapping() {
+  sair.program {
+    %0 = sair.static_range 8 : !sair.range
+    %1 = sair.alloc[d0:%0] : !sair.value<d0:range, memref<f32>>
+    %2 = sair.placeholder : !sair.range
+    %3 = sair.placeholder[d0:%2] : !sair.range<d0:range>
+    sair.free[d0:%2, d1:%3] %1(unstripe(d0, d1, [4]))
+      : !sair.value<d0:range x d1:range(d0), memref<f32>>
+    sair.exit
+  }
+  return
+}