@@ -252,23 +252,23 @@ def incsubtensor_rv_replace(fgraph, node):
252252
253253logprob_rewrites_db = SequenceDB ()
254254logprob_rewrites_db .name = "logprob_rewrites_db"
255- logprob_rewrites_db .register ("pre-canonicalize" , optdb .query ("+canonicalize" ), - 10 , "basic" )
255+ logprob_rewrites_db .register ("pre-canonicalize" , optdb .query ("+canonicalize" ), "basic" )
256256
257257# These rewrites convert un-measurable variables into their measurable forms,
258258# but they need to be reapplied, because some of the measurable forms require
259259# their inputs to be measurable.
260260measurable_ir_rewrites_db = NoCallbackEquilibriumDB ()
261261measurable_ir_rewrites_db .name = "measurable_ir_rewrites_db"
262262
263- logprob_rewrites_db .register ("measurable_ir_rewrites" , measurable_ir_rewrites_db , - 10 , "basic" )
263+ logprob_rewrites_db .register ("measurable_ir_rewrites" , measurable_ir_rewrites_db , "basic" )
264264
265265# These rewrites push random/measurable variables "down", making them closer to
266266# (or eventually) the graph outputs. Often this is done by lifting other `Op`s
267267# "up" through the random/measurable variables and into their inputs.
268- measurable_ir_rewrites_db .register ("subtensor_lift" , local_subtensor_rv_lift , - 5 , "basic" )
269- measurable_ir_rewrites_db .register ("incsubtensor_lift" , incsubtensor_rv_replace , - 5 , "basic" )
268+ measurable_ir_rewrites_db .register ("subtensor_lift" , local_subtensor_rv_lift , "basic" )
269+ measurable_ir_rewrites_db .register ("incsubtensor_lift" , incsubtensor_rv_replace , "basic" )
270270
271- logprob_rewrites_db .register ("post-canonicalize" , optdb .query ("+canonicalize" ), 10 , "basic" )
271+ logprob_rewrites_db .register ("post-canonicalize" , optdb .query ("+canonicalize" ), "basic" )
272272
273273
274274def construct_ir_fgraph (
0 commit comments