CheckParameterValues not replaced in inner graphs · pymc-devs/pymc#6697 | Good First Issue

(3 评论) (0 反应) (0 负责人)Python (1,902 fork)batch import

help wantedlogprob

仓库指标

Star: (7,926 star)
PR 合并指标: (平均合并 11天) (30 天内合并 7 个 PR)

描述

We include these rewrites when compiling logp graphs, but this won't replace inner logp graphs (e.g, inside Scan or not-inlined OpFromGraph)

https://github.com/pymc-devs/pymc/blob/371472dcf1c44f99a9ad4e7317606404b2a47a0e/pymc/pytensorf.py#L1142-L1144

The example in #6696 suffers from this. Sampling is quite faster when the rewrite is manually added

贡献者指南

研究方向: 问题在于 CheckParameterValues 重写未应用于内部图（例如，Scan 内部或未内联的 OpFromGraph）。相关代码位于 pymc/pytensorf.py 第 1142-1144 行。修复需要确保此重写递归应用于所有子图。首先检查编译过程中重写如何应用，并查看 issue #6696 中的示例以复现性能差异。考虑修改重写应用逻辑以遍历内部图。
技术栈: python
领域: backend
议题类型: 缺陷
难度: 3
预计时间: 半天
活动状态: 活跃
清晰度: 清晰
前置要求: Pythonbasic PyMC knowledge
新手友好度: 55