[Mlir-commits] [mlir] [mlir][tosa] Add row_gather_block_scaled op (PR #192272)

[Luke Hutton]

================
@@ -2526,6 +2526,44 @@ def Tosa_GatherOp : Tosa_InferShapedTypeOp<"gather", [NoMemoryEffect]> {
       "operands attr-dict `:` functional-type(operands, results)";
 }
 
+//===----------------------------------------------------------------------===//
+// Operator: row_gather_block_scaled
+//===----------------------------------------------------------------------===//
+def Tosa_RowGatherBlockScaledOp
+    : Tosa_InferShapedTypeOp<"row_gather_block_scaled", [NoMemoryEffect]> {
+  let summary =
+      "Row gather operation for block-scaled and non-block-scaled data.";
+
+  let description = [{
+    Generate a tensor-list which contains a data tensor and an optional scale
+    tensor based on the input indices and row_count. The number of consecutive
+    rows gathered for each index is specified in row_count.
----------------
lhutton1 wrote:

I think it would be worth indicating what the variadic of values means here (similar to reshape_block_scaled above)

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