File tree Expand file tree Collapse file tree 1 file changed +0
-26
lines changed
jaxlib/mosaic/dialect/tpu/transforms Expand file tree Collapse file tree 1 file changed +0
-26
lines changed Original file line number Diff line number Diff 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);
You can’t perform that action at this time.
0 commit comments