Skip to content

Commit fcefb07

Browse files
committed
fix naming in optimization
1 parent 8ef63b9 commit fcefb07

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

third_party/amd/lib/TritonAMDGPUTransforms/OptimizeBufferOpPtr.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)