Skip to content

Commit 5a6756d

Browse files
[mlir][AMGPU] Replace use of SmallVector with ArrayRef, NFC (#163770)
Improving choice of class used, from SmallVector to ArrayRef (https://llvm.org/docs/ProgrammersManual.html#llvm-adt-arrayref-h). Also infer template types when possible. Leftover from llvm/llvm-project#155951. --------- Signed-off-by: Muzammiluddin Syed <[email protected]>
1 parent 919395b commit 5a6756d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

mlir/lib/Dialect/AMDGPU/IR/AMDGPUDialect.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -757,13 +757,13 @@ struct PackScales final : OpRewritePattern<ScaledMFMAOp> {
757757
offset = numElements - 4l;
758758
}
759759
Type scaleSrcElemType = scaleSrcType.getElementType();
760-
auto newSrcType = VectorType::get(SmallVector<int64_t>({numElements}),
761-
scaleSrcElemType);
760+
auto newSrcType =
761+
VectorType::get(ArrayRef{numElements}, scaleSrcElemType);
762762
Value newScaleSrc =
763763
vector::ShapeCastOp::create(rewriter, loc, newSrcType, scaleSrc);
764764
auto extract = vector::ExtractStridedSliceOp::create(
765-
rewriter, loc, newScaleSrc, ArrayRef<int64_t>{offset},
766-
ArrayRef<int64_t>{size}, ArrayRef<int64_t>{1});
765+
rewriter, loc, newScaleSrc, ArrayRef{offset}, ArrayRef{size},
766+
ArrayRef{int64_t(1)});
767767
rewriter.modifyOpInPlace(op, [&] {
768768
op->setOperand(opIdx, extract);
769769
setOpsel(opIdx, opsel);

0 commit comments

Comments
 (0)