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

(3 comments) (0 reactions) (0 assignees)Python (1,902 forks)batch import

help wantedlogprob

Repository metrics

Stars: (7,926 stars)
PR merge metrics: (平均マージ 11d) (30d で 7 merged PRs)

説明

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
Issue 種別: バグ
難度: 3
推定時間: 半日
活動状況: アクティブ
明確さ: 明確
前提条件: Pythonbasic PyMC knowledge
初心者向け度: 55