pymc-devs/pymc

Only sample using a single thread when sampling multiple chains in parallel

已關閉

#5,453 建立於 2022年2月9日

在 GitHub 查看貢獻者指南
 (1 則留言) (0 個反應) (0 位負責人)Python (1,902 個分叉)batch import
help wantedrequest discussionv4

倉庫指標

星標
 (7,926 顆星)
PR 合併指標
 (平均合併 11天) (30 天內合併 7 個 PR)

描述

My prior of what cores=4 does when calling pm.sample() is that pymc would use 4 of my cores. However, because of multi-threading this might not be the case. Furthermore, multi-threading is causing problems when running multiple chains in parallel: https://github.com/pymc-devs/pymc/issues/3140

I thus think we should set os.environ['OMP_NUM_THREADS'] = '1' if we're sampling on > 1 core.

貢獻者指南

研究方向
在 pm.sample() 中設定 os.environ['OMP NUM THREADS'] = '1'，以避免在取樣多個鏈時出現多執行緒衝突。查看 issue #3140 了解背景。確保單鏈取樣不受影響。
技術棧
python
領域
backend
議題類型
錯誤
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
2
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-3 小時
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
Pythonpymc
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
70