+ j `aR t 0tRt^RIt^RIHtR .tRRltRRlt/t]^kRRlt R#) aShared helpers for native ops backed by nvmath-python / cuSOLVER. Centralizes nvmath availability checks, cuSOLVER version gating, and cuSOLVER handle management so multiple native ops (e.g. linalg.polar, and upcoming matmul-family ops) don't each reimplement them. All functions here are import-light at module load (nvmath is only imported inside the function bodies, lazily, on first use) so importing this module does not pull in the nvmath/cuda runtime or initialize CUDA. N)_unavailable_reasonc2V^8dQhR\R,/#returnN)str)formats"[/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_native/nvmath_utils.py __annotate__r s--3:-c \\4#)zReturn a human-readable reason nvmath is unavailable, or None if present. Does not import nvmath (uses importlib.util.find_spec under the hood). )r _NVMATH_DEPSr r nvmath_unavailable_reasonrs | ,,r c2V^8dQhR\R,/#r)int)rs"r r r s#*r cT^RIHpVP4# \dR#i;i)acuSOLVER runtime version as major*1000 + minor*100 + patch, or None. Queries cusolverGetVersion through nvmath, against the exact cuSOLVER library nvmath dlopens (which is whatever the loader resolves -- the system CUDA install or the nvidia-cu* wheels a torch build depends on). The encoded value does not include the build number, so it can gate out runtimes that lack a symbol entirely but cannot distinguish patch builds. cusolverN)nvmath.bindingsr get_version Exceptionrs r cusolver_versionrs+,##%% s  ''cDV^8dQhR\PR\/#)rdevicer)torchrr)rs"r r r :s    r c^RIHp\PV4pVfJ\P P V4;_uu_4VP4pRRR4V\V&V# +'giL;i)z7Return a cached cuSOLVER Dn handle bound to ``device``.) cusolverDnN)rr_handlesgetrcudarcreate)rcshandles& r get_cusolver_handler$:sU0 \\& !F ~ ZZ  v & &YY[F'! M' &s A++ A; cvV^8dQh/^\9d%\\P\3,;R&#)rr)__conditional_annotations__dictrrr)rs"r r r s.  l'&$u||S !&m r )z nvmath-pythonznvmath.bindings) r&__doc__r common_utilsrr rrrr$r )r&s@r r*s8  - ) -0%'& r