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