Skip to content

Commit 417dc61

Browse files
authored
[Backend] Fix build due to mid air collision (#7651)
triton-lang/triton#7636 changed function signature but triton-lang/triton#7460 wasn't picking it up before landing.
1 parent ebad1d9 commit 417dc61

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

lib/Conversion/TritonGPUToLLVM/Utility.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2049,8 +2049,8 @@ void finalizeTensorAtomicResults(Operation *op, RankedTensorType tensorTy,
20492049
{{str_attr("offset"), dstLayout.getTotalOutDimSize()}});
20502050
auto smemBase = LLVM::getSharedMemoryBase(loc, rewriter, targetInfo, op);
20512051

2052-
auto emitSt = [&](ConversionPatternRewriter &rewriter, Location loc,
2053-
ArrayRef<Value> vals, Value shmemAddr, int idx,
2052+
auto emitSt = [&](RewriterBase &rewriter, Location loc, ArrayRef<Value> vals,
2053+
Value shmemAddr, int idx,
20542054
VectorType vecTy) -> SmallVector<Value> {
20552055
auto length = vecTy.getNumElements();
20562056
Value valsVec =
@@ -2060,8 +2060,8 @@ void finalizeTensorAtomicResults(Operation *op, RankedTensorType tensorTy,
20602060
return {};
20612061
};
20622062

2063-
auto emitLd = [&](ConversionPatternRewriter &rewriter, Location loc,
2064-
ArrayRef<Value> vals, Value shmemAddr, int idx,
2063+
auto emitLd = [&](RewriterBase &rewriter, Location loc, ArrayRef<Value> vals,
2064+
Value shmemAddr, int idx,
20652065
VectorType vecTy) -> SmallVector<Value> {
20662066
Value loadedVec = targetInfo.loadDShared(rewriter, loc, shmemAddr,
20672067
std::nullopt, vecTy, b.true_val());

0 commit comments

Comments
 (0)