pytensor 3.0.5__tar.gz → 3.0.7__tar.gz

{pytensor-3.0.5/pytensor.egg-info → pytensor-3.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytensor
-Version: 3.0.5
+Version: 3.0.7
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License-Expression: BSD-3-Clause

{pytensor-3.0.5 → pytensor-3.0.7}/pytensor/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2026-06-11T03:57:58+0200",
+ "date": "2026-06-15T16:18:33+0200",
  "dirty": false,
  "error": null,
- "full-revisionid": "3411b300d7a120e7fb58b5ad5495391ddd74e189",
- "version": "3.0.5"
+ "full-revisionid": "cb44be3432acef912fa74cb068620d011901f564",
+ "version": "3.0.7"
 }
 '''  # END VERSION_JSON
 

{pytensor-3.0.5 → pytensor-3.0.7}/pytensor/tensor/rewriting/indexed_elemwise.py

@@ -6,7 +6,6 @@ so the Numba backend can generate a single loop with indirect indexing,
 eliminating materialised intermediate arrays.
 """
 
-from pytensor.compile import optdb
 from pytensor.compile.builders import OpFromGraph
 from pytensor.graph import node_rewriter
 from pytensor.graph.rewriting.basic import GraphRewriter, dfs_rewriter
@@ -170,17 +169,23 @@ def undo_take_reshape_for_fusion(fgraph, node):
 
 
 indexed_elemwise_optdb = SequenceDB()
-optdb.register(
-    "fuse_indexed_into_elemwise",
-    indexed_elemwise_optdb,
-    "numba",
-    # symbolic_op_recognition is excluded from OpFromGraph inner-graph
-    # compilation, preventing recursive fusion.
-    "symbolic_op_recognition",
-    # After inplace_elemwise (position=50.5) so we see final inplace patterns,
-    # same position as other numba-specific rewrites (BlockwiseWithCoreShape).
-    position=100,
-)
+# Disabled by default on the 3.0.x line. This fusion shipped in 3.0.4 but can
+# trigger a RecursionError during OpFromGraph shape inference on some graphs
+# (e.g. numba-mode FullRankADVI); it was too risky for a minor release. The
+# proper fix lives on main (non-recursive core-shape construction). Until that
+# is backported we do not register the fusion into the default numba pipeline.
+# To re-enable, re-add `from pytensor.compile import optdb` and the call below.
+# optdb.register(
+#     "fuse_indexed_into_elemwise",
+#     indexed_elemwise_optdb,
+#     "numba",
+#     # symbolic_op_recognition is excluded from OpFromGraph inner-graph
+#     # compilation, preventing recursive fusion.
+#     "symbolic_op_recognition",
+#     # After inplace_elemwise (position=50.5) so we see final inplace patterns,
+#     # same position as other numba-specific rewrites (BlockwiseWithCoreShape).
+#     position=100,
+# )
 
 indexed_elemwise_optdb.register(
     "undo_take_dimshuffle_for_fusion",

{pytensor-3.0.5 → pytensor-3.0.7/pytensor.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytensor
-Version: 3.0.5
+Version: 3.0.7
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License-Expression: BSD-3-Clause