even faster on pimcomp models

2026-07-31 21:15:28 +02:00
parent 9ca1a0ed9f
commit f4a3b012cc
49 changed files with 1923 additions and 583 deletions
@@ -118,6 +118,20 @@ LogicalResult PimCoreBatchOp::verify() {
  return verifyOnlyConstantExternalValues(getOperation(), getBody(), "pim.core_batch");
 }

+LogicalResult PimVMVOp::verify() {
+  if (getLength() <= 0)
+    return emitError("length must be positive");
+  if (failed(verifyCompatibleShapedTypes(
+        getOperation(), getTarget().getType(), getOutput().getType(), "target and output types must match")))
+    return failure();
+  auto targetType = cast<ShapedType>(getTarget().getType());
+  auto sourceType = dyn_cast<ShapedType>(getSource().getType());
+  if (!sourceType || !haveSameShapedContainerKind(getTarget().getType(), getSource().getType())
+      || targetType.getElementType() != sourceType.getElementType())
+    return emitError("target and source must use the same shaped container kind and element type");
+  return success();
+}
+
 LogicalResult PimVMMOp::verify() {
  if (failed(verifyCompatibleShapedTypes(
        getOperation(), getOutputBuffer().getType(), getOutput().getType(), "output buffer and output must match")))