Merge with fast resnet

2026-07-20 18:01:58 +02:00
parent 5c94e00f43 ab54243fda
commit 961fd613bd
85 changed files with 4475 additions and 4455 deletions
@@ -193,6 +193,14 @@ forEachContiguousDestinationChunk(ArrayRef<int64_t> destShape,
  return visit(visit, 0);
 }

+int64_t getFragmentAssemblySourceElementOffset(RankedTensorType sourceType,
+                                               int64_t sourceSlot,
+                                               int64_t sourceOffset) {
+  assert(sourceType.getRank() > 0 && sourceType.hasStaticShape()
+         && "fragment assembly source must have a static leading slot dimension");
+  return sourceSlot * (sourceType.getNumElements() / sourceType.getDimSize(0)) + sourceOffset;
+}
+
 static mlir::Value
 createSteppedOffset(OpBuilder& builder, Location loc, mlir::Value start, mlir::Value index,
                    int64_t stepBytes, Operation *constantAnchor) {