Skip to content

Commit 7461aff

Browse files
chsiggtensorflower-gardener
authored andcommitted
Fix mlir cast/dyn_cast/isa.
Use llvm::cast/dyn_cast/isa free functions since members are deprecated in llvm/llvm-project#135556 PiperOrigin-RevId: 749476645
1 parent 3bec0ad commit 7461aff

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow/compiler/mlir/quantization/stablehlo/passes/tf_insert_weight_param.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -219,7 +219,7 @@ class InsertWeightParamPattern
219219
dimension_numbers.getRhsContractingDimensions();
220220
ArrayRef<int64_t> rhs_batching_dims =
221221
dimension_numbers.getRhsBatchingDimensions();
222-
int64_t rank = dot.getRhs().getType().cast<TensorType>().getRank();
222+
int64_t rank = cast<TensorType>(dot.getRhs().getType()).getRank();
223223
for (int i = 0; i < rank; ++i) {
224224
// Return the first non-contracting, non-batching dimension of rhs.
225225
if (llvm::find(rhs_contracting_dims, i) == rhs_contracting_dims.end() &&
@@ -228,7 +228,7 @@ class InsertWeightParamPattern
228228
}
229229
}
230230
}
231-
return op.getOperand(1).getType().cast<TensorType>().getRank() - 1;
231+
return cast<TensorType>(op.getOperand(1).getType()).getRank() - 1;
232232
}
233233
};
234234

0 commit comments

Comments
 (0)