Use llvm::cast/dyn_cast/isa since alternatives are deprecated in llvm/llvm-project#135556

PiperOrigin-RevId: 749057356

Author: TFRT team

include/tfrt/basic_kernels/opdefs/basic_kernels.td

@@ -89,8 +89,8 @@ class ConstantOp<string suffix, Type baseType, Attr attr>
 def F16Attr : FloatAttrBase<F16, "16-bit float attribute">;
 def BF16Attr : TypedAttrBase<
   BF16, "FloatAttr",
-  And<[CPred<"$_self.isa<::mlir::FloatAttr>()">,
-       CPred<"$_self.cast<::mlir::FloatAttr>().getType().isBF16()">]>,
+  And<[CPred<"llvm::isa<::mlir::FloatAttr>($_self)">,
+       CPred<"llvm::cast<::mlir::FloatAttr>($_self).getType().isBF16()">]>,
   "bfloat16 attribute"> {
   let returnType = [{ ::llvm::APFloat }];
 }

include/tfrt/basic_kernels/opdefs/tfrt_base.td

@@ -38,19 +38,19 @@ def TFRT_Dialect : Dialect {
 }
 
 def TFRT_ChainType :
-    Type<CPred<"$_self.isa<tfrt::compiler::ChainType>()">, "!tfrt.chain type">,
+    Type<CPred<"llvm::isa<tfrt::compiler::ChainType>($_self)">, "!tfrt.chain type">,
     BuildableType<"$_builder.getType<tfrt::compiler::ChainType>()">;
 
 def TFRT_StringType :
-    Type<CPred<"$_self.isa<tfrt::compiler::StringType>()">, "!tfrt.string type">,
+    Type<CPred<"llvm::isa<tfrt::compiler::StringType>($_self)">, "!tfrt.string type">,
     BuildableType<"$_builder.getType<tfrt::compiler::StringType>()">;
 
 def TFRT_TensorTypeType :
-    Type<CPred<"$_self.isa<tfrt::compiler::TensorTypeType>()">, "!tfrt.tensor_type type">,
+    Type<CPred<"llvm::isa<tfrt::compiler::TensorTypeType>($_self)">, "!tfrt.tensor_type type">,
     BuildableType<"$_builder.getType<tfrt::compiler::TensorTypeType>()">;
 
 def TFRT_DeviceType :
-    Type<CPred<"$_self.isa<tfrt::compiler::DeviceType>()">, "!tfrt.device type">,
+    Type<CPred<"llvm::isa<tfrt::compiler::DeviceType>($_self)">, "!tfrt.device type">,
     BuildableType<"$_builder.getType<tfrt::compiler::DeviceType>()">;
 
 #endif  // TFRT_BASE

include/tfrt/core_runtime/opdefs/corert_base.td

@@ -35,12 +35,12 @@ def CoreRT_Dialect : Dialect {
 // Deivce is an aggregate of Core Runtime ops. It is passed to corert.executeop
 // so that the required Core Runtime can be looked up in this device.
 def CoreRT_OpHandlerType :
-    Type<CPred<"$_self.isa<tfrt::corert::OpHandlerType>()">, "!corert.ophandler type">,
+    Type<CPred<"isa<tfrt::corert::OpHandlerType>($_self)">, "!corert.ophandler type">,
     BuildableType<"$_builder.getType<tfrt::corert::OpHandlerType>()">;
 
 // TensorHandle is the currency type for Core Runtime execution.
 def CoreRT_TensorHandleType:
-    Type<CPred<"$_self.isa<tfrt::corert::TensorHandleType>()">, "!corert.tensorhandle type">,
+    Type<CPred<"isa<tfrt::corert::TensorHandleType>($_self)">, "!corert.tensorhandle type">,
     BuildableType<"$_builder.getType<tfrt::corert::TensorHandleType>()">;
 
 #endif

include/tfrt/tensor/opdefs/host_tensor.td

@@ -43,7 +43,7 @@ def HostTensor_Dialect : Dialect {
 // Type definitions
 //===----------------------------------------------------------------------===//
 def HostBufferType : DialectType<HostTensor_Dialect,
-    CPred<"$_self.isa<::tfrt::ht::HostBufferType>()">, "!ht.host_buffer type">,
+    CPred<"llvm::isa<::tfrt::ht::HostBufferType>($_self)">, "!ht.host_buffer type">,
     BuildableType<"$_builder.getType<::tfrt::ht::HostBufferType>()"> {
   let description = [{
     `!ht.host_buffer type` represents a host buffer.

include/tfrt/tensor/opdefs/tensor.td

@@ -42,7 +42,7 @@ def Tensor_Dialect : Dialect {
 // Type definitions
 //===----------------------------------------------------------------------===//
 def TensorType : DialectType<Tensor_Dialect,
-    CPred<"$_self.isa<::tfrt::tfrt_tensor::TensorType>()">, "!tfrt_tensor.tensor type">,
+    CPred<"llvm::isa<::tfrt::tfrt_tensor::TensorType>($_self)">, "!tfrt_tensor.tensor type">,
     BuildableType<"$_builder.getType<::tfrt::tfrt_tensor::TensorType>()"> {
   let description = [{
     `!trft_tensor.tensor type` represents a generic tfrt tensor.

include/tfrt/tensor/opdefs/tensor_shape_base.td

@@ -39,15 +39,15 @@ def TensorShapeDialect : Dialect {
 
 // Type definition.
 def TS_Shape : DialectType<TensorShapeDialect,
-    CPred<"$_self.isa<::tfrt::ts::ShapeType>()">, "!ts.shape type">,
+    CPred<"llvm::isa<::tfrt::ts::ShapeType>($_self)">, "!ts.shape type">,
     BuildableType<"$_builder.getType<::tfrt::ts::ShapeType>()"> {
   let description = [{
     `!ts.shape type` represents a static tensor shape.
   }];
 }
 
 def TS_PartialShape : DialectType<TensorShapeDialect,
-    CPred<"$_self.isa<::tfrt::ts::PartialShapeType>()">, "!ts.partial_shape type">,
+    CPred<"llvm::isa<::tfrt::ts::PartialShapeType>($_self)">, "!ts.partial_shape type">,
     BuildableType<"$_builder.getType<::tfrt::ts::PartialShapeType>()"> {
   let description = [{
     `!ts.partial_shape type` represents either a static tensor shape, unranked

lib/bef_converter/mlir_to_bef/bef_location_emitter.cc

@@ -28,7 +28,7 @@ bool BefLocationEmitter::IsSupportedLocation(const mlir::Location& loc) {
   if (llvm::isa<mlir::UnknownLoc>(loc)) return true;
   if (llvm::isa<mlir::NameLoc>(loc)) return true;
   if (llvm::isa<mlir::FileLineColLoc>(loc)) return true;
-  if (auto callsite_loc = loc.dyn_cast<mlir::CallSiteLoc>()) {
+  if (auto callsite_loc = llvm::dyn_cast<mlir::CallSiteLoc>(loc)) {
     return IsSupportedLocation(callsite_loc.getCallee()) &&
            IsSupportedLocation(callsite_loc.getCaller());
   }