Skip to content

Commit 467e496

Browse files
authored
Support mkl_fft in virtual environment out of the box (#140)
* Support mkl_fft in virtual environment out of the box
1 parent 69defde commit 467e496

File tree

2 files changed

+34
-0
lines changed

2 files changed

+34
-0
lines changed

Diff for: mkl_fft/__init__.py

+3
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
2525
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2626

27+
from . import _init_helper
2728
from ._pydfti import (fft, ifft, fft2, ifft2, fftn, ifftn, rfftpack, irfftpack,
2829
rfft, irfft, rfft2, irfft2, rfftn, irfftn)
2930

@@ -32,3 +33,5 @@
3233

3334
__all__ = ['fft', 'ifft', 'fft2', 'ifft2', 'fftn', 'ifftn', 'rfftpack', 'irfftpack'
3435
'rfft', 'irfft', 'rfft2', 'irfft2', 'rfftn', 'irfftn','interfaces']
36+
37+
del _init_helper

Diff for: mkl_fft/_init_helper.py

+31
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)