Refactor + ReduceMean batched

2026-05-29 15:57:13 +02:00
parent 832bd7f1f7
commit 819d8af0f7
27 changed files with 929 additions and 568 deletions
@@ -113,8 +113,8 @@ static void materializeHostConstantsInCore(CoreOpTy coreOp,
            rewriter,
            op->getLoc(),
            originalType,
-            getOrCreateHostIndexConstant(op, 0, constantFolder),
-            getOrCreateHostIndexConstant(op, static_cast<int64_t>(resolvedAddress->byteOffset), constantFolder),
+            getOrCreateHostIndexConstant(constantFolder, op, 0),
+            getOrCreateHostIndexConstant(constantFolder, op, static_cast<int64_t>(resolvedAddress->byteOffset) ),
            deviceDst,
            getGlobalOp.getResult(),
            rewriter.getI32IntegerAttr(static_cast<int32_t>(totalBytes)))