o TdT!@sdZddlZddlZddlZddlZddZeddlmZddlm Z edZ e e Z [[ dd l m Z dd l mZmZdd lmZmZmZmZmZmZmZmZdd lTddlmmZdd lTdd lmZmZmZm Z m!Z!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+ddl,m-Z-m.Z.m/Z/ddl0m1Z1ddl2Z0ddl2m3Z4ddl2m5Z5ddl6Z0ddl7Z0ddZ8d9ej:ej:Z:dZ;dZe=ddl?Z? e>e _@dS)z5 Expose top-level symbols that are safe for import * NcCsdd}tjdd}|dkr d|dd|d d}t|ddl}||}|d kr=d |dd|d d}t||d krEtd zddl}Wn tyUYdSw||}|dkrid|jd}t|dS)a Make sure the Python, NumPy and SciPy present are supported versions. This has to be done _before_ importing anything from Numba such that incompatible versions can be reported to the user. If this occurs _after_ importing things from Numba and there's an issue in e.g. a Numba c-ext, a SystemError might have occurred which prevents reporting the likely cause of the problem (incompatible versions of critical dependencies). cSsttt|jdddS)N.)tuplemapint __version__split)modr a/home/ncw/WWW/www-new/content/articles/pi-bbp/venv/lib/python3.10/site-packages/numba/__init__.pyextract_versionsz._ensure_critical_deps..extract_versionNr)z.Numba needs Python 3.8 or greater. Got Python rr)rz-Numba needs NumPy 1.21 or greater. Got NumPy )rzNumba needs NumPy 1.24 or less)rrz7Numba requires SciPy version 1.0 or greater. Got SciPy )sys version_info ImportErrornumpyscipyr)r PYVERSIONmsgnp numpy_versionr sp_versionr r r _ensure_critical_deps s@   rr) get_versions)generate_version_infoversion)config)typeserrors)typeofprangepndindexgdbgdb_breakpointgdb_init literallyliteral_unroll)*)cfunc generated_jitjitnjitstencil jit_module) vectorize guvectorizethreading_layerget_num_threadsset_num_threadsset_parallel_chunksizeget_parallel_chunksize get_thread_id)carrayfarray from_dtype) experimental)objmode_context)parallel_chunksizecKsddlm}|j|fi|S)Nr) _runtests) numba.testingr@main)argvkwdsruntestsr r r testss rFaK cfunc from_dtype guvectorize jit experimental njit stencil jit_module typeof prange gdb gdb_breakpoint gdb_init vectorize objmode literal_unroll get_num_threads set_num_threads set_parallel_chunksize get_parallel_chunksize parallel_chunksize )r(r)rrcCsddl}ddl}td}||j}|r/ttt| }|t kr.dt |jf}t |n| dddl m}m}|tkrJdt|}t ||dS)z, Make sure llvmlite is operational. rNz(\d+)\.(\d+).(\d+)zfNumba requires at least version %d.%d.%d of llvmlite. Installed version is %s. Please update llvmlite.z'llvmlite version format not recognized!)llvm_version_infocheck_jit_executionzNumba requires at least version %d.%d.%d of LLVM. Installed llvmlite is built against version %d.%d.%d. Please update llvmlite.)warningsllvmliterecompilematchrrrrgroups_min_llvmlite_versionrwarnllvmlite.bindingrIrJ_min_llvm_version)rKrLregexmverrrIrJr r r _ensure_llvms(     rXcCstjs|z^tjdrtjdntjdrtjdntjdr+tjdnWdSzttjj ds;WWdSWnt yWt d krStj rSd }t |YWdSwtjd d Wd St d krvtj ryt dYdSYdSYdSdS)zU Tries to enable SVML if configuration permits use and the library is found. linuxz libsvml.sodarwinz libsvml.dylibwin svml_dispmdFhas_svmlx86_64z>SVML was found but llvmlite >= 0.23.2 is needed to support it.SVMLz-vector-library=SVMLTz'SVML was not found/could not be loaded.)r DISABLE_INTEL_SVMLrplatform startswithrLbindingload_library_permanentlygetattrtargetsAttributeErrormachineDEBUGrKrR set_option)rr r r _try_enable_svmls:        rk)A__doc__rarMrrKr_versionrnumba.misc.init_utilsrrr numba.corer r!r"numba.misc.specialr#r$r%r&r'r(r)r*numba.core.errorsnumba.core.typescorenumba.core.decoratorsr,r-r.r/r0r1numba.np.ufuncr2r3r4r5r6r7r8r9numba.np.numpy_supportr:r;r<numbar=numba.core.withcontextsr>objmoder?numba.core.target_extension numba.typedrFr__all__rQrTrXrkrL USING_SVMLr r r r sT*    ( (   #/