From df799cdefae1e46682296a4e1d3f0d298112da87 Mon Sep 17 00:00:00 2001
From: DawnARC <dawnarc@luminarc.tech>
Date: Mon, 4 May 2026 17:50:22 +0800
Subject: [PATCH] =?UTF-8?q?fix:=E6=9B=B4=E6=AD=A3=E9=BB=98=E8=AE=A4?=
 =?UTF-8?q?=E5=80=BC=EF=BC=8C=E9=81=BF=E5=85=8D=E6=84=8F=E6=96=99=E4=B9=8B?=
 =?UTF-8?q?=E5=A4=96=E7=9A=84=E5=86=85=E5=AD=98=E5=8D=A0=E7=94=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

用 os.environ.setdefault 将原生库线程数（OMP_NUM_THREADS、OPENBLAS_NUM_THREADS、MKL_NUM_THREADS、NUMEXPR_NUM_THREADS）的默认值设置为 "1"。这样可以确保导入的代码默认不会过度占用 CPU 线程（以及巨量内存）
---
 src/__init__.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/__init__.py b/src/__init__.py
index e69de29b..fd2235c7 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -0,0 +1,14 @@
+from __future__ import annotations
+
+import os
+
+
+_NATIVE_THREAD_ENV_DEFAULTS = {
+    "OMP_NUM_THREADS": "1",
+    "OPENBLAS_NUM_THREADS": "1",
+    "MKL_NUM_THREADS": "1",
+    "NUMEXPR_NUM_THREADS": "1",
+}
+
+for _name, _value in _NATIVE_THREAD_ENV_DEFAULTS.items():
+    os.environ.setdefault(_name, _value)