Better implementation of MergeComputeNodesPass

2026-07-15 10:40:14 +02:00
parent 51fdb830e5
commit c744f388dc
26 changed files with 4892 additions and 4217 deletions
@@ -68,6 +68,23 @@ Value insertStaticSlice(
    .getResult();
 }

+Value extractMixedSliceOrIdentity(RewriterBase &rewriter,
+                                  Location loc,
+                                  Value source,
+                                  RankedTensorType resultType,
+                                  const MixedSliceGeometry &geometry) {
+  return extractStaticSliceOrIdentity(rewriter, loc, source, resultType,
+                                      geometry.offsets, geometry.sizes,
+                                      geometry.strides);
+}
+
+Value insertMixedSlice(OpBuilder &builder, Location loc, Value source,
+                       Value dest, const MixedSliceGeometry &geometry) {
+  return tensor::InsertSliceOp::create(builder, loc, source, dest,
+                                       geometry.offsets, geometry.sizes,
+                                       geometry.strides);
+}
+
 FailureOr<Value> addLeadingUnitTensorDimension(OpBuilder& builder, Location loc, Value value) {
  auto type = dyn_cast<RankedTensorType>(value.getType());
  if (!type || !type.hasStaticShape())