Skip to content

Commit e257fed

Browse files
apaszkeGoogle-ML-Automation
authored andcommitted
[Mosaic TPU] Remove debug-assert-insertion from tpu.td
PiperOrigin-RevId: 834706417
1 parent 8276789 commit e257fed

File tree

2 files changed

+1
-12
lines changed

2 files changed

+1
-12
lines changed

jaxlib/mosaic/dialect/tpu/tpu.td

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1468,16 +1468,7 @@ def TPU_LogBufferOp : TPU_Op<"log_buffer"> {
14681468
let hasVerifier = 1;
14691469
}
14701470

1471-
def DebugAssertInsertionPass : Pass<"debug-assert-insertion", "::mlir::func::FuncOp"> {
1472-
let dependentDialects = [
1473-
"::mlir::func::FuncDialect",
1474-
"::mlir::arith::ArithDialect",
1475-
"::mlir::cf::ControlFlowDialect",
1476-
"::mlir::vector::VectorDialect",
1477-
"::mlir::tpu::TPUDialect",
1478-
];
1479-
let constructor = "::mlir::tpu::createDebugAssertInsertionPass()";
1480-
}
1471+
14811472

14821473
def LogicalToPhysicalDeviceIdPass : Pass<"logical-to-physical-device-id", "::mlir::func::FuncOp"> {
14831474
let dependentDialects = [

jaxlib/mosaic/dialect/tpu/tpu_dialect.h

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,8 +98,6 @@ createPreCanonicalizationOptimizationPass(
9898
std::unique_ptr<OperationPass<func::FuncOp>>
9999
createLogicalToPhysicalDeviceIdPass(int64_t total_devices);
100100

101-
std::unique_ptr<OperationPass<func::FuncOp>> createDebugAssertInsertionPass();
102-
103101
#define GEN_PASS_DECL_MOSAICSERDEPASS
104102
#include "jaxlib/mosaic/dialect/tpu/tpu_passes.h.inc"
105103

0 commit comments

Comments
 (0)