File size: 358 Bytes
20c251e | 1 2 3 4 5 6 7 8 9 | from __future__ import annotations
import os
# Login nodes and small CI runners often impose strict process/thread limits. Keep numerical tests
# deterministic and leave production training thread counts to the launcher configuration.
for variable in ("OMP_NUM_THREADS", "OPENBLAS_NUM_THREADS", "MKL_NUM_THREADS"):
os.environ.setdefault(variable, "1")
|