big refactor

2026-08-04 11:28:05 +02:00
parent f4a3b012cc
commit 10b6ee6c32
150 changed files with 6737 additions and 4816 deletions
@@ -280,12 +280,12 @@ static FailureOr<Value> buildReduceMeanKeepdimsBlueprint(
  SmallVector<int64_t> fragmentStrides(fragmentOffsets.size(), 1);
  return spatial::SpatBlueprintOp::create(
           rewriter, loc, keepdimsType, batchValue, ValueRange {},
-           rewriter.getStringAttr("nchw"),
-           rewriter.getStringAttr("fragmented"),
+           spatial::getNCHWLayout(rewriter.getContext()),
+           spatial::getFragmentedLayout(rewriter.getContext()),
           rewriter.getDenseI64ArrayAttr(fragmentOffsets),
           rewriter.getDenseI64ArrayAttr(fragmentSizes),
           rewriter.getStringAttr("reduce_mean_keepdims_fragments"),
-           rewriter.getStringAttr("fragment_assembly"),
+           spatial::getFragmentAssemblyMode(rewriter.getContext()),
           rewriter.getDenseI64ArrayAttr(operandIndices),
           rewriter.getDenseI64ArrayAttr(sourceSlots),
           rewriter.getDenseI64ArrayAttr(sourceOffsets),