Skip to content

Commit 937bb29

Browse files
committed
Serialize OpenBLAS workers for parallel ViZDoom forks on Linux
1 parent a673eaf commit 937bb29

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

envpool/vizdoom/BUILD

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,11 @@ py_test(
168168
name = "vizdoom_pretrain_test",
169169
size = "large",
170170
srcs = ["vizdoom_pretrain_test.py"],
171+
# NumPy and OpenCV OpenBLAS fork hooks race while Doom starts in parallel.
172+
env = select({
173+
"//:linux": {"OPENBLAS_NUM_THREADS": "1"},
174+
"//conditions:default": {},
175+
}),
171176
imports = ["../.."],
172177
tags = ["exclusive"],
173178
deps = [

0 commit comments

Comments
 (0)