Skip to content

Commit

Permalink
Feedback
Browse files Browse the repository at this point in the history
  • Loading branch information
nsmithtt committed Aug 28, 2024
1 parent 210d236 commit 3027752
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions lib/Dialect/TTMetal/Transforms/Passes.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -420,10 +420,11 @@ class TTIRToTTMetalDispatchRewriter : public OpRewritePattern<ttir::GenericOp> {
return AffineMap::get(physShape.size(), 0, resultExpr, memref.getContext());
}

Value i32(std::int32_t value, OpBuilder &b) const {
return b
.create<arith::ConstantOp>(b.getUnknownLoc(), b.getI32Type(),
b.getI32IntegerAttr(value))
Value i32(std::int32_t value, OpBuilder &builder) const {
return builder
.create<arith::ConstantOp>(builder.getUnknownLoc(),
builder.getI32Type(),
builder.getI32IntegerAttr(value))
.getResult();
}

Expand Down Expand Up @@ -479,7 +480,7 @@ class TTIRToTTMetalDispatchRewriter : public OpRewritePattern<ttir::GenericOp> {
for (BlockArgument operand : blockArguments) {
auto cbType = mlir::cast<ttkernel::CBType>(operand.getType());
AffineMap affineIterator = getAffineIterator(cbType.getMemref());
auto *match = iteratorMaps.find(affineIterator);
auto match = iteratorMaps.find(affineIterator);
assert(match != iteratorMaps.end());
blockArgIteratorMapping[operand.getArgNumber()] =
std::distance(iteratorMaps.begin(), match);
Expand Down

0 comments on commit 3027752

Please sign in to comment.