@@ -108,7 +108,7 @@ struct AdvanceBasePointer : public OpRewritePattern<scf::ForOp> {
108108 // - offsetInitializer is a value of offset on first loop iteration
109109 // - incrementOp is an operation that advances offset tensor
110110 struct BufferOpInfo {
111- amdttg::BufferOpAddressinInterface op;
111+ amdttg::BufferOpInterface op;
112112 Value offsetIncrement;
113113 Value baseIncrement;
114114 Value offsetInitializer;
@@ -118,7 +118,7 @@ struct AdvanceBasePointer : public OpRewritePattern<scf::ForOp> {
118118 // Perform series of checks to decide if given operation could be optimized.
119119 // If optimization is possible, return filled BufferOpInfo
120120 static std::optional<BufferOpInfo>
121- analyzeBufferOp (amdttg::BufferOpAddressinInterface op, scf::ForOp targetFor) {
121+ analyzeBufferOp (amdttg::BufferOpInterface op, scf::ForOp targetFor) {
122122 LDBG (" Analyzing: " << *op);
123123 Value maybeOffsetsBlockArg = op.getOffsets ();
124124 auto maybeOffsetDefOp = maybeOffsetsBlockArg.getDefiningOp ();
@@ -257,7 +257,7 @@ struct AdvanceBasePointer : public OpRewritePattern<scf::ForOp> {
257257 // Replace base ptr with incrementing value
258258 for (auto [info, basePtr, nextBasePtr] :
259259 llvm::zip (infoList, basePtrs, nextIterBasePtrs)) {
260- auto newBufferOp = cast<amdttg::BufferOpAddressinInterface >(
260+ auto newBufferOp = cast<amdttg::BufferOpInterface >(
261261 mapping.lookup <Operation *>(info.op .getOperation ()));
262262 newBufferOp.getOffsetsMutable ().assign (info.offsetInitializer );
263263 // two cases:
@@ -272,7 +272,7 @@ struct AdvanceBasePointer : public OpRewritePattern<scf::ForOp> {
272272
273273 static SmallVector<BufferOpInfo> collectBufferOps (scf::ForOp forOp) {
274274 SmallVector<BufferOpInfo> list;
275- forOp.walk ([&list, forOp](amdttg::BufferOpAddressinInterface op) {
275+ forOp.walk ([&list, forOp](amdttg::BufferOpInterface op) {
276276 auto info = analyzeBufferOp (op, forOp);
277277 if (info.has_value ()) {
278278 list.push_back (info.value ());
0 commit comments