| #ifndef TPU_ATTRS |
| #define TPU_ATTRS |
| |
| include "third_party/py/jax/jaxlib/mosaic/dialect/tpu/tpu.td" |
| include "mlir/Pass/PassBase.td" |
| |
| def LowerToLLOPass : Pass<"lower-to-llo", "::mlir::func::FuncOp"> { |
| let dependentDialects = [ |
| "::mlir::arith::ArithDialect", |
| "::mlir::cf::ControlFlowDialect", |
| "::mlir::func::FuncDialect", |
| "::mlir::math::MathDialect", |
| "::mlir::memref::MemRefDialect", |
| "::mlir::scf::SCFDialect", |
| "::mlir::vector::VectorDialect", |
| "::mlir::llo::LLODialect", |
| "::mlir::tpu::TPUDialect", |
| ]; |
| let constructor = "::mlir::tpu::createPartialLowerToLLOPass()"; |
| let options = [ |
| Option<"mock_target", "mock-target", "int", /*default=*/"-1", "">, |
| ]; |
| } |
| |
| #endif // TPU_ATTRS |
