diff --git a/lib/Analysis/AxisInfo.cpp b/lib/Analysis/AxisInfo.cpp index eb29b0c63307..f0c5ae3167ec 100644 --- a/lib/Analysis/AxisInfo.cpp +++ b/lib/Analysis/AxisInfo.cpp @@ -895,7 +895,7 @@ class ShLIOpAxisInfoVisitor final : public BinaryOpVisitorImpl { lhsDivisibility = 1; } auto numBits = log2Int(lhsDivisibility); - return multiplyDivisor(lhsDivisibility, 1 << shift); + return multiplyDivisor(lhsDivisibility, 1ll << shift); } int64_t getConstancy(arith::ShLIOp op, const AxisInfo &lhs, diff --git a/lib/Conversion/TritonGPUToLLVM/ScanOpToLLVM.cpp b/lib/Conversion/TritonGPUToLLVM/ScanOpToLLVM.cpp index 19572444f322..c69a6b32425e 100644 --- a/lib/Conversion/TritonGPUToLLVM/ScanOpToLLVM.cpp +++ b/lib/Conversion/TritonGPUToLLVM/ScanOpToLLVM.cpp @@ -76,7 +76,7 @@ static void warpScan(SmallVector> &srcValues, acc[j] = select(mask, tempAcc[j], acc[j]); } } - srcValues[srcIndex] = acc; + srcValues[srcIndex] = std::move(acc); } } @@ -128,8 +128,8 @@ static void AddPartialReduce(SmallVector> &srcValues, ConversionPatternRewriter &rewriter, const TargetInfoBase &targetInfo, ScanLoweringHelper &helper, - SmallVector smemBases, - SmallVector smemTypes, Value warpId, + ArrayRef smemBases, + ArrayRef smemTypes, Value warpId, Value laneIdAxis, Value parallelLaneId) { Location loc = helper.getLoc(); unsigned numParallelLane = helper.getNonAxisNumThreadsPerCTA(); @@ -224,7 +224,7 @@ static void AddPartialReduce(SmallVector> &srcValues, srcValues[srcIndex - i * elementStride][j]); } } - srcValues[srcIndex - i * elementStride] = laneValue; + srcValues[srcIndex - i * elementStride] = std::move(laneValue); } // For the next chunk start back from the value containing the // accumulated value of all the warps. @@ -303,7 +303,7 @@ static void AddPartialReduceOneWarp(SmallVector> &srcValues, srcValues[srcIndex - i * elementStride][j], laneValue[j]); } } - srcValues[srcIndex - i * elementStride] = laneValue; + srcValues[srcIndex - i * elementStride] = std::move(laneValue); } // For the next chunk start back from the value containing the // accumulated value of all the warps.