[Mlir-commits] [mlir] [mlir][tosa] Add TOSA RESHAPE_BLOCK_SCALED support (PR #191149)

[Tai Ly]

================
@@ -2270,6 +2270,44 @@ def Tosa_ReshapeOp : Tosa_InferTensorTypeOp<"reshape", [Pure]> {
   let assemblyFormat = "operands attr-dict `:` functional-type(operands, results)";
 }
 
+//===----------------------------------------------------------------------===//
+// Operator: reshape_block_scaled
+//===----------------------------------------------------------------------===//
+def Tosa_ReshapeBlockScaledOp
+    : Tosa_InferTensorTypeOp<"reshape_block_scaled", [Pure]> {
+  let summary = "Reshape with support for block scaled tensors.";
+
+  let description = [{
+    Returns a tensor-list with the same type/values as the input, with a new
----------------
Tai78641 wrote:

There is no description that says this operator allow only either
one tensor (for the non-block-scaled case) or two tensors (for the block-scaled case, in which case, the second tensor must be a scale tensor for the first).
This is enforced in verifier.
The variadic $input is misleading, as if this op could reshape multiple unrelated tensors at once.

https://github.com/llvm/llvm-project/pull/191149