restore Dot22 rewrites

Author: dehorsley

pytensor/tensor/rewriting/linalg.py

@@ -2,6 +2,7 @@
 
 from pytensor.graph.rewriting.basic import node_rewriter
 from pytensor.tensor import basic as at
+from pytensor.tensor.blas import Dot22
 from pytensor.tensor.elemwise import DimShuffle
 from pytensor.tensor.math import Dot, Prod, dot, log
 from pytensor.tensor.math import pow as at_pow
@@ -31,12 +32,12 @@ def transinv_to_invtrans(fgraph, node):
 
 
 @register_stabilize
-@node_rewriter([Dot])
+@node_rewriter([Dot, Dot22])
 def inv_as_solve(fgraph, node):
     """
     This utilizes a boolean `symmetric` tag on the matrices.
     """
-    if isinstance(node.op, Dot):
+    if isinstance(node.op, (Dot, Dot22)):
         l, r = node.inputs
         if l.owner and isinstance(l.owner.op, MatrixInverse):
             return [solve(l.owner.inputs[0], r)]
@@ -122,7 +123,7 @@ def cholesky_ldotlt(fgraph, node):
         return
 
     A = node.inputs[0]
-    if not (A.owner and isinstance(A.owner.op, Dot)):
+    if not (A.owner and isinstance(A.owner.op, (Dot, Dot22))):
         return
 
     l, r = A.owner.inputs