o i!@sdZddlZddlZddlZddlZddZeddlmZddlm Z edZ e e Z [[ dd l m Z dd l mZmZdd lmZmZmZmZmZmZmZmZdd lTddlmmZdd lTdd lmZmZmZm 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.ddl/m0Z0ddl1Z/ddl1m2Z3ddl1m4Z4ddl5Z/ddl6Z/ddZ7d8ej9ej9Z9dZ:dZ;ddZZ> 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 krRd |dd|d d}t|zddl}Wn tybYdSw||}|dkrvd|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 L/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/__init__.pyextract_versionsz._ensure_critical_deps..extract_versionNr) z/Numba needs Python 3.10 or greater. Got Python rr)rz-Numba needs NumPy 1.22 or greater. Got NumPy )rz)Numba needs NumPy 2.4 or less. Got NumPy )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 sJ   rr) get_versions)generate_version_infoversion)config)typeserrors)typeofprangepndindexgdbgdb_breakpointgdb_init literallyliteral_unroll)*)cfuncjitnjitstencil 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 testws rEaK 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.bindingrHrI_min_llvm_version)rJrKregexmverrrHrIr r r _ensure_llvms(     rWcCstjs|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 startswithrKbindingload_library_permanentlygetattrtargetsAttributeErrormachineDEBUGrJrQ set_option)rr r r _try_enable_svmls:        rj)@__doc__r`rLrrJr_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/r0numba.np.ufuncr1r2r3r4r5r6r7r8numba.np.numpy_supportr9r:r;numbar<numba.core.withcontextsr=objmoder>numba.core.target_extension numba.typedrEr__all__rPrSrWrjrK USING_SVMLr r r r sT.    ((   #/