Skip to content

Commit 13e721a

Browse files
tlongeriGoogle-ML-Automation
authored andcommitted
[Mosaic:TPU][NFC] Delete unused functions
PiperOrigin-RevId: 707660214
1 parent 74eca13 commit 13e721a

File tree

1 file changed

+0
-26
lines changed

1 file changed

+0
-26
lines changed

jaxlib/mosaic/dialect/tpu/transforms/infer_vector_layout.cc

Lines changed: 0 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -1968,32 +1968,6 @@ class VectorLayoutInferer {
19681968
setOutLayout(op, out);
19691969
}
19701970

1971-
SmallVector<Layout, 4> getInLayout(Operation *op) {
1972-
CHECK(op);
1973-
CHECK(op->getAttr("in_layout"));
1974-
auto in_attrs = op->getAttrOfType<ArrayAttr>("in_layout").getValue();
1975-
CHECK_EQ(in_attrs.size(), op->getNumOperands());
1976-
SmallVector<Layout, 4> in_layouts;
1977-
in_layouts.reserve(op->getNumOperands());
1978-
for (int i = 0; i < op->getNumOperands(); ++i) {
1979-
in_layouts.push_back(cast<VectorLayoutAttr>(in_attrs[i]).getLayout());
1980-
}
1981-
return in_layouts;
1982-
}
1983-
1984-
SmallVector<Layout, 4> getOutLayout(Operation *op) {
1985-
CHECK(op);
1986-
CHECK(op->getAttr("out_layout"));
1987-
auto out_attrs = op->getAttrOfType<ArrayAttr>("out_layout").getValue();
1988-
CHECK_EQ(out_attrs.size(), op->getNumResults());
1989-
SmallVector<Layout, 4> out_layouts;
1990-
out_layouts.reserve(op->getNumResults());
1991-
for (int i = 0; i < op->getNumResults(); ++i) {
1992-
out_layouts.push_back(cast<VectorLayoutAttr>(out_attrs[i]).getLayout());
1993-
}
1994-
return out_layouts;
1995-
}
1996-
19971971
Layout getLayout(Value v) {
19981972
auto op = v.getDefiningOp();
19991973
CHECK(op);

0 commit comments

Comments
 (0)