update deprecated AffineExpr casts

nod-ai · Nov 17, 2023 · 7b31dcf · 7b31dcf
1 parent bcf89d2
commit 7b31dcf
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/compiler/src/iree/compiler/Codegen/LLVMCPU/LLVMCPUFoldVectorContractUnitDims.cpp b/compiler/src/iree/compiler/Codegen/LLVMCPU/LLVMCPUFoldVectorContractUnitDims.cpp
@@ -54,7 +54,7 @@ dropFoldableUnitIndices(PatternRewriter &rewriter,
     SmallVector<int64_t> dstShape;
     SmallVector<AffineExpr> dstExpr;
     for (const auto &expr : enumerate(map.getResults())) {
-      if (auto dimExpr = expr.value().dyn_cast<AffineDimExpr>()) {
+      if (auto dimExpr = llvm::dyn_cast<AffineDimExpr>(expr.value())) {
         if (!foldableDims.contains(dimExpr.getPosition())) {
           dstShape.push_back(contractShape[dimExpr.getPosition()]);
           unsigned numSkipped = 0;

diff --git a/compiler/src/iree/compiler/Codegen/LLVMCPU/SetSpecialTilingConfigs.cpp b/compiler/src/iree/compiler/Codegen/LLVMCPU/SetSpecialTilingConfigs.cpp
@@ -37,7 +37,7 @@ static void setTileSizes(linalg::GenericOp intMatmul,
   int out = 2;
   auto hasDim = [&](int mapIdx, int dimIdx) -> bool {
     return llvm::any_of(maps[mapIdx].getResults(), [&](AffineExpr res) {
-      auto expr = res.dyn_cast<AffineDimExpr>();
+      auto expr = llvm::dyn_cast<AffineDimExpr>(res);
       return expr && expr.getPosition() == dimIdx;
     });
   };

diff --git a/compiler/src/iree/compiler/Codegen/LLVMCPU/VectorContractCustomKernels.cpp b/compiler/src/iree/compiler/Codegen/LLVMCPU/VectorContractCustomKernels.cpp
@@ -147,7 +147,7 @@ static bool isVectorTimesMatrixTransposed(vector::ContractionOp contractionOp,
     }
     for (int r = 0; r < expectedResults.size(); ++r) {
       int actualMapResult =
-          map.getResults()[r].cast<AffineDimExpr>().getPosition();
+          llvm::cast<AffineDimExpr>(map.getResults()[r]).getPosition();
       if (actualMapResult != expectedMapResults[m][r]) {
         return false;
       }

diff --git a/compiler/src/iree/compiler/Preprocessing/Common/ConvertConvToChannelsLast.cpp b/compiler/src/iree/compiler/Preprocessing/Common/ConvertConvToChannelsLast.cpp
@@ -260,11 +260,11 @@ static TransposeIndices collectChannelTransposeIndices(
     AffineMap map, SmallVector<SmallVector<unsigned, 2>> transposeDimTargets) {
   SmallVector<TransposeIndices> channelIndices(transposeDimTargets.size());
   for (auto [index, result] : llvm::enumerate(map.getResults())) {
-    if (result.isa<AffineDimExpr>()) {
+    if (llvm::isa<AffineDimExpr>(result)) {
       for (auto [channelVec, dimCategory] :
            llvm::zip_equal(channelIndices, transposeDimTargets)) {
         if (llvm::is_contained(dimCategory,
-                               result.cast<AffineDimExpr>().getPosition())) {
+                               llvm::cast<AffineDimExpr>(result).getPosition())) {
           channelVec.push_back(index);
           break;
         }