diff --git a/aesara/tensor/math_opt.py b/aesara/tensor/math_opt.py index bc83e33808..efebb8dd5c 100644 --- a/aesara/tensor/math_opt.py +++ b/aesara/tensor/math_opt.py @@ -3532,6 +3532,8 @@ def local_reciprocal_1_plus_exp(fgraph, node): return out +# x_lv, if any, is the logic variable version of some term x, +# while x_at, if any, is the Aesara tensor version for the same. def distribute_mul_over_add(in_lv, out_lv): from kanren import conso, eq, fact, heado, tailo @@ -3696,7 +3698,7 @@ def distribute_div_over_sub(in_lv, out_lv): fastmath = EquilibriumDB() -optdb.register("fastmath", fastmath, 0.05, "fast_run", "mul") +optdb.register("fastmath", fastmath, 1, "fast_run", "mul") fastmath.register( "dist_mul_over_add_opt", diff --git a/tests/tensor/test_math_opt.py b/tests/tensor/test_math_opt.py index 1d0bd0742a..c995126318 100644 --- a/tests/tensor/test_math_opt.py +++ b/tests/tensor/test_math_opt.py @@ -4556,6 +4556,8 @@ class TestDistributiveOpts: y_at = vector("y") a_at = matrix("a") + # x_lv, if any, is the logic variable version of some term x, + # while x_at is the Aesara tensor version for the same. @pytest.mark.parametrize( "orig_operation, optimized_operation", [