wehub-resource-sync
28558dca80
Build / Build and test on ubuntu-24.04-arm for aarch64 (push) Waiting to run
Build / Build and test on windows-11-arm for aarch64 (push) Waiting to run
Build / Build and test on macos-latest for x86_64 (push) Waiting to run
Build / Build and test on windows-latest for x86_64 (push) Waiting to run
Build / Build and test on ubuntu-latest for x86_64 (push) Failing after 1s
Build / Build and test on ubuntu-latest (numpy 1.26) for x86_64 (push) Failing after 0s
67 lines
2.3 KiB
Python
67 lines
2.3 KiB
Python
###################################################################
|
|
# Numexpr - Fast numerical array expression evaluator for NumPy.
|
|
#
|
|
# License: MIT
|
|
# Author: See AUTHORS.txt
|
|
#
|
|
# See LICENSE.txt and LICENSES/*.txt for details about copyright and
|
|
# rights to use.
|
|
####################################################################
|
|
|
|
"""
|
|
Numexpr is a fast numerical expression evaluator for NumPy. With it,
|
|
expressions that operate on arrays (like "3*a+4*b") are accelerated
|
|
and use less memory than doing the same calculation in Python.
|
|
|
|
See:
|
|
|
|
https://github.com/pydata/numexpr
|
|
|
|
for more info about it.
|
|
|
|
"""
|
|
|
|
from numexpr.interpreter import __BLOCK_SIZE1__, MAX_THREADS, use_vml
|
|
|
|
is_cpu_amd_intel = False # DEPRECATION WARNING: WILL BE REMOVED IN FUTURE RELEASE
|
|
|
|
# cpuinfo imports were moved into the test submodule function that calls them
|
|
# to improve import times.
|
|
|
|
from numexpr.expressions import E
|
|
from numexpr.necompiler import (NumExpr, disassemble, evaluate, re_evaluate,
|
|
validate)
|
|
from numexpr.utils import (_init_num_threads, detect_number_of_cores,
|
|
detect_number_of_threads, get_num_threads,
|
|
get_vml_version, set_num_threads,
|
|
set_vml_accuracy_mode, set_vml_num_threads)
|
|
|
|
# Detect the number of cores
|
|
ncores = detect_number_of_cores()
|
|
# Initialize the number of threads to be used
|
|
nthreads = _init_num_threads()
|
|
# The default for VML is 1 thread (see #39)
|
|
# set_vml_num_threads(1)
|
|
|
|
from . import version
|
|
|
|
__version__ = version.version
|
|
|
|
def print_versions():
|
|
"""Print the versions of software that numexpr relies on."""
|
|
try:
|
|
import numexpr.tests
|
|
return numexpr.tests.print_versions()
|
|
except ImportError:
|
|
# To maintain Python 2.6 compatibility we have simple error handling
|
|
raise ImportError('`numexpr.tests` could not be imported, likely it was excluded from the distribution.')
|
|
|
|
def test(verbosity=1):
|
|
"""Run all the tests in the test suite."""
|
|
try:
|
|
import numexpr.tests
|
|
return numexpr.tests.test(verbosity=verbosity)
|
|
except ImportError:
|
|
# To maintain Python 2.6 compatibility we have simple error handling
|
|
raise ImportError('`numexpr.tests` could not be imported, likely it was excluded from the distribution.')
|