| #ifndef PLATFORMS_XLA_MOSAIC_CUSTOM_CALL_KERNEL_NAME_H_ |
| #define PLATFORMS_XLA_MOSAIC_CUSTOM_CALL_KERNEL_NAME_H_ |
| |
| #include <string> |
| |
| #include "third_party/absl/status/statusor.h" |
| #include "third_party/llvm/llvm-project/mlir/include/mlir/IR/Operation.h" |
| |
| namespace xla::mosaic { |
| |
| // Returns the 'kernel_name' contained in the given op. |
| // Returns an error if the string cannot be found. |
| absl::StatusOr<std::string> CustomCallKernelName(mlir::Operation* op); |
| |
| } // namespace xla::mosaic |
| |
| #endif // PLATFORMS_XLA_MOSAIC_CUSTOM_CALL_KERNEL_NAME_H_ |
