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

[Luke Hutton]

================
@@ -3326,6 +3338,48 @@ LogicalResult tosa::GatherOp::inferReturnTypeComponents(
   return success();
 }
 
+LogicalResult tosa::RowGatherBlockScaledOp::inferReturnTypeComponents(
+    MLIRContext *context, ::std::optional<Location> location,
+    RowGatherBlockScaledOp::Adaptor adaptor,
+    SmallVectorImpl<ShapedTypeComponents> &inferredReturnShapes) {
+  const auto values = adaptor.getValues();
+  if (values.empty() || values.size() > 2)
----------------
lhutton1 wrote:

this should be enforced by the verifier - I think we can remove it

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