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

[Luke Hutton]

================
@@ -114,6 +114,17 @@ static LogicalResult checkConstantOperandMatMul(Operation *op,
   return success();
 }
 
+static LogicalResult
+checkConstantOperandRowGatherBlockScaled(Operation *op, const TargetEnv &env) {
+  if (!env.allows(Extension::dynamic) &&
+      isa<tosa::RowGatherBlockScaledOp>(op)) {
+    auto rowGatherOp = cast<tosa::RowGatherBlockScaledOp>(op);
+    const unsigned rowCountIndex = rowGatherOp.getValues().size() + 1;
+    return checkConstantOperands(op, {rowCountIndex});
+  }
+  return success();
+}
+
----------------
lhutton1 wrote:

If we're planning to have this operation fully validated, we should also add this operation to level checks. Since the operation is experimental, I'm happy to leave it out if lack of level checking is indicated in the commit message.

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