pi@UddlZddlZddlZddlmZddlmZddlZddlm Z m Z m Z m Z m Z mZ ddlmZdaeeed<edZndZd Zej.ej0ej2hZd Zd Z dd Ze dd ZGddeZeej@e!e!ej@e!<e"ed<e"ed<e"ed<e"ed<eed<y#e$rdZYwxYw)N)contextmanager)Optional)__allow_nonbracketed_mutation_FP32Precision_get_fp32_precision_getter_set_fp32_precision_setter ContextProp PropModule)_cudnn__cudnn_versionc0 ttjatj}tj}|\}}}|\}}}||k7rd}n |dkstj s||k(}n||k\}|st jjdddk(ryd|d|d }d t jvrRt jjd d  t fd d Drt|d|t|dt|y)NF&PYTORCH_SKIP_CUDNN_COMPATIBILITY_CHECK01Tz=cuDNN version incompatibility: PyTorch was compiled against z but found runtime version z. PyTorch already comes bundled with cuDNN. One option to resolving this error is to ensure PyTorch can find the bundled cuDNN. LD_LIBRARY_PATHc3&K|]}|v ywN).0 substringld_library_paths _/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/torch/backends/cudnn/__init__.py z_init..As9B _4s)cudacudnnzLooks like your LD_LIBRARY_PATH contains incompatible version of cudnn. Please either remove it from the path or install cudnn zHone possibility is that there is a conflicting cuDNN in LD_LIBRARY_PATH.) r r getVersionIntgetRuntimeVersiongetCompileVersionis_cudaosenvirongetany RuntimeError) runtime_versioncompile_version runtime_major runtime_minor_ compile_major compile_minorcudnn_compatiblebase_error_msgrs @r_initr0"sU  "$224O$668O$668O.= +M=!.= +M=! -#( "&..#0M#A #0M#A #::>>"JCPTWW55D4EF11@0AB34% 2&(jjnn5F&KOFW+-.VVeUfh +-.DE '~66cy)NFrrr1rr0r0Vsr1c$tsytS)zReturn the version of cuDNN.N)r0r rr1rversionr4Zs 7 r1c6tjjS)z9Return a bool indicating if CUDNN is currently available.)torch_C _has_cudnnrr1r is_availabler9hs 88  r1c|tjjsy|jjdk7s|j t vrytstjdytsGtjdjdddjtjd yy ) NFrzPyTorch was compiled without cuDNN/MIOpen support. To use cuDNN/MIOpen, rebuild PyTorch making sure the library is visible to the build system.z4cuDNN/MIOpen library not found. Check your {libpath}DYLD_LIBRARY_PATHPATH)darwinwin32r)libpathT)r6r7_get_cudnn_enableddevicetypedtypeCUDNN_TENSOR_DTYPESr9warningswarnr0formatr$sysplatform)tensors r is_acceptablerKms 88 & & ( }}V#v||;N'N >  N  7 B I I#6HLLLL"3 J   r1c &tjjtjjt sdntjj tjj tjjtjjddf}|tjj||tjj||)t rtjj||tjj||tjj||!tjjdd||S)Nrall)r6r7r@_get_cudnn_benchmarkr9_cuda_get_cudnn_benchmark_limit_get_cudnn_deterministic_get_cudnn_allow_tf32r_set_cudnn_enabled_set_cudnn_benchmark_cuda_set_cudnn_benchmark_limit_set_cudnn_deterministic_set_cudnn_allow_tf32r)_enabled _benchmark_benchmark_limit_deterministic _allow_tf32_fp32_precision orig_flagss r set_flagsr^s ##% %%' N(P(P(R ))+ &&( ++FE: J ##H- %%j1#  001AB! )).9 &&{3" ++FE?K r1c #Kt5t||||||}ddd dt5tdddy#1swY*xYw#1swYyxYw#t5tdddw#1swYwxYwxYwwr)rr^)enabled benchmarkbenchmark_limit deterministic allow_tf32fp32_precisionr]s rflagsrfs ' (          # + , # z " # #   # # * , # z " # # #s_ B AB A B  A B A B AB  B* A<3 B<B BB cleZdZfdZeej jej jZ eej jej jZ eej jej jZdZer1eej j$ej j&Zeej j(ej j*ZeddZeddZeeddeddZxZS) CudnnModulec&t|||yr)super__init__)selfmname __class__s rrkzCudnnModule.__init__s D!r1NrconvrnnrM)__name__ __module__ __qualname__rkr r6r7r@rRr`rPrUrcrNrSrarbr9rOrTrQrVrdrrprqrrre __classcell__)ros@rrhrhs"%((55uxx7R7RSG ))588+L+LM %%uxx'D'DIO~% HH 4 4 HH 4 4  &&(F(FJ && )D  'C "651"651Nr1rhr`rcrardrb)NNNNNnone)FF FTrv)#r"rHrE contextlibrtypingrr6torch.backendsrrrrr r torch._Cr ImportErrorr int__annotations__r0r4halffloatdoublerDr9rKr^rfrhmodulesrrboolrr1rrs % "&#% 0h JJ KK LL 0 >  ##<*<$CKK$98D H    e FsCCC