summaryrefslogtreecommitdiff
path: root/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
diff options
context:
space:
mode:
authorMehdi Amini <joker.eph@gmail.com>2023-09-12 01:12:22 -0700
committerMehdi Amini <joker.eph@gmail.com>2023-09-12 01:33:47 -0700
commit830b9b072d8458ee89c48f00d4de59456c9f467f (patch)
treed1afe474578dd720e6de2ef9bb956793df3f2ce8 /mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
parent0a8d17e79b02a92814a2a788d79df1f54d70ec3e (diff)
Update some uses of `getAttr()` to be explicit about Inherent vs Discardable (NFC)
Diffstat (limited to 'mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp')
-rw-r--r--mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
index 370d8bb19364..2fbf1babf437 100644
--- a/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
+++ b/mlir/test/lib/Dialect/Vector/TestVectorTransforms.cpp
@@ -681,7 +681,8 @@ struct TestCreateVectorBroadcast
auto targetShape =
cast<VectorType>(op->getResult(0).getType()).getShape();
auto arrayAttr =
- cast<DenseI64ArrayAttr>(op->getAttr("broadcast_dims")).asArrayRef();
+ cast<DenseI64ArrayAttr>(op->getDiscardableAttr("broadcast_dims"))
+ .asArrayRef();
llvm::SetVector<int64_t> broadcastedDims;
broadcastedDims.insert(arrayAttr.begin(), arrayAttr.end());
OpBuilder b(op);