Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions mkl/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

import sys
from . import _init_helper


class RTLD_for_MKL:
Expand Down Expand Up @@ -57,3 +58,5 @@ def __exit__(self, *args):

from ._py_mkl_service import *
from ._version import __version__

del _init_helper
31 changes: 31 additions & 0 deletions mkl/_init_helper.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
# Copyright 2025 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import os
import os.path
import sys

is_venv_win32 = (
sys.platform == "win32"
and sys.base_exec_prefix != sys.exec_prefix
and os.path.isfile(os.path.join(sys.exec_prefix, "pyvenv.cfg"))
)

if is_venv_win32:
# In Windows venv: add Library/bin to PATH for proper DLL loading
dll_dir = os.path.join(sys.exec_prefix, "Library", "bin")
if os.path.isdir(dll_dir):
os.add_dll_directory(dll_dir)

del is_venv_win32
Loading