@@ -2049,8 +2049,8 @@ void finalizeTensorAtomicResults(Operation *op, RankedTensorType tensorTy,
2049
2049
{{str_attr (" offset" ), dstLayout.getTotalOutDimSize ()}});
2050
2050
auto smemBase = LLVM::getSharedMemoryBase (loc, rewriter, targetInfo, op);
2051
2051
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,
2054
2054
VectorType vecTy) -> SmallVector<Value> {
2055
2055
auto length = vecTy.getNumElements ();
2056
2056
Value valsVec =
@@ -2060,8 +2060,8 @@ void finalizeTensorAtomicResults(Operation *op, RankedTensorType tensorTy,
2060
2060
return {};
2061
2061
};
2062
2062
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,
2065
2065
VectorType vecTy) -> SmallVector<Value> {
2066
2066
Value loadedVec = targetInfo.loadDShared (rewriter, loc, shmemAddr,
2067
2067
std::nullopt, vecTy, b.true_val ());
0 commit comments