Commits on Sep 6, 2024

1. [shape_polyO] Improve handling of equality shape constraints …

   This fixes several bugs in presence of equality constraints where
   the left-hand side is just a dimension variable.
   
   First, such constraints were not applied when parsing variables.
   Now, with a constraint `a == b` when we parse "a" we obtain `b`.
   
   Second, when we evaluate symbolic dimensions that contain
   dimension variables that are constrained to be equal to something
   else, we may fail to find the dimension variable in the environment
   because the environment construction has applied the constraints.
   We fix this by looking up the unknown dimension variable in
   the equality constraints.
   
   Fixes: jax-ml#23437
   Fixes: jax-ml#23456

   

   gnecula committed Sep 6, 2024
   Configuration menu

   • View commit details
   • Copy full SHA for 0d8ffd3
   • Browse repository at this point

   Copy the full SHA

   0d8ffd3 View commit details

   Browse the repository at this point in the history