Skip to content

Commit 334abcd

Browse files
committed
Support mkl-service in virtual environment out of the box
1 parent 2c23623 commit 334abcd

File tree

2 files changed

+34
-0
lines changed

2 files changed

+34
-0
lines changed

mkl/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2525

2626
import sys
27+
from . import _init_helper
2728

2829

2930
class RTLD_for_MKL:
@@ -57,3 +58,5 @@ def __exit__(self, *args):
5758

5859
from ._py_mkl_service import *
5960
from ._version import __version__
61+
62+
del _init_helper

mkl/_init_helper.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
# Copyright 2025 Intel Corporation
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# http://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
import os
16+
import os.path
17+
import sys
18+
19+
is_venv_win32 = (
20+
sys.platform == "win32"
21+
and sys.base_exec_prefix != sys.exec_prefix
22+
and os.path.isfile(os.path.join(sys.exec_prefix, "pyvenv.cfg"))
23+
)
24+
25+
if is_venv_win32:
26+
# In Windows venv: add Library/bin to PATH for proper DLL loading
27+
dll_dir = os.path.join(sys.exec_prefix, "Library", "bin")
28+
if os.path.isdir(dll_dir):
29+
os.add_dll_directory(dll_dir)
30+
31+
del is_venv_win32

0 commit comments

Comments
 (0)