Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

PKG: Suppress error when two version of OpenMP are loaded #58

Merged
merged 1 commit into from
Aug 30, 2024
Merged

PKG: Suppress error when two version of OpenMP are loaded #58

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
19 changes: 19 additions & 0 deletions src/diptest/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,11 +1,30 @@
from __future__ import annotations

import importlib.metadata
import os

from diptest.diptest import dipstat, diptest
from diptest.lib import _diptest_core as _diptest
from diptest.lib._diptest_core import _has_openmp_support

# Setting the following environment variable allows multiple OpenMP
# libraries to be loaded. This is also used without issue by Scikit-learn.
# OpenMP error msg:
# /* OMP: Error #15: Initializing libomp.dylib, but found libomp.dylib already initialized.
# * OMP: Hint This means that multiple copies of the OpenMP runtime have been linked into the program.
# * That is dangerous, since it can degrade performance or cause incorrect results.
# * The best thing to do is to ensure that only a single OpenMP runtime is linked into the process,
# * e.g. by avoiding static linking of the OpenMP runtime in any library.
# * As an unsafe, unsupported, undocumented workaround you can set the environment variable KMP_DUPLICATE_LIB_OK=TRUE
# * to allow the program to continue to execute, but that may cause crashes or silently produce incorrect results.
# * For more information, please see http://openmp.llvm.org/
# */
os.environ.setdefault("KMP_DUPLICATE_LIB_OK", "True")

# Workaround issue discovered in intel-openmp 2019.5:
# https://github.com/ContinuumIO/anaconda-issues/issues/11294
os.environ.setdefault("KMP_INIT_AT_FORK", "FALSE")

__version__ = importlib.metadata.version("diptest")

__all__ = ["dipstat", "diptest", "_diptest", "_has_openmp_support", "__version__"]
Loading