Commita418956

authored and

committed

[Mosaic TPU] Remove the relayout insertion pass from tpu.td

PiperOrigin-RevId: 837541620

1 parent42d0f9b commita418956Copy full SHA for a418956

File tree

-20

lines changed

-20

lines changed

Lines changed: 0 additions & 16 deletions

Original file line number	Diff line number	Diff line change
`@@ -1569,22 +1569,6 @@ def InferVectorLayoutPass : Pass<"tpu-infer-vector-layout", "::mlir::func::FuncO`
`1569`	`1569`	`];`
`1570`	`1570`	`}`
`1571`	`1571`
`1572`		`-def RelayoutInsertionPass : Pass<"tpu-relayout-insertion", "::mlir::func::FuncOp"> {`
`1573`		`- let dependentDialects = [`
`1574`		`- "::mlir::arith::ArithDialect",`
`1575`		`- "::mlir::func::FuncDialect",`
`1576`		`- "::mlir::tpu::TPUDialect",`
`1577`		`- ];`
`1578`		`- let constructor = "::mlir::tpu::createRelayoutInsertionPass()";`
`1579`		`- let options = [`
`1580`		`- // If hardware_generation is not set, the default value of -1 will crash on`
`1581`		`- // runOnOperation.`
`1582`		`- Option<"hardware_generation", "hardware-generation", "int", /default=/"-1", "">,`
`1583`		`- Option<"lane_count", "lane-count", "int", /default=/"128", "">,`
`1584`		`- Option<"sublane_count", "sublane-count", "int", /default=/"8", "">,`
`1585`		`- ];`
`1586`		`-}`
`1587`		`-`
`1588`	`1572`	`def ApplyVectorLayoutPass : Pass<"tpu-apply-vector-layout", "::mlir::func::FuncOp"> {`
`1589`	`1573`	`let dependentDialects = [`
`1590`	`1574`	`"::mlir::arith::ArithDialect",`

Lines changed: 0 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -83,10 +83,6 @@ std::unique_ptr<OperationPass<func::FuncOp>> createInferVectorLayoutPass(`
`83`	`83`	`const TpuTilingFlags& tpu_tiling_flags = {},`
`84`	`84`	`bool shape_invariant_numerics =true);`
`85`	`85`
`86`		`-std::unique_ptr<OperationPass<func::FuncOp>>createRelayoutInsertionPass(`
`87`		`-int hardware_generation = -1,`
`88`		`- std::array<int64_t,2> target_shape = {8,128});`
`89`		`-`
`90`	`86`	`std::unique_ptr<OperationPass<func::FuncOp>>createApplyVectorLayoutPass(`
`91`	`87`	`const ApplyVectorLayoutContext &ctx = ApplyVectorLayoutContext{});`
`92`	`88`

Comments

(0)