§
    Û×j   ã                   ó   — d Z ddlZd„ ZdS )zJUse bundled CUDA/HIP helpers for the validated Windows Triton stacks only.é    Nc                  ó2  ‡	‡
‡‡‡‡‡‡‡‡‡‡‡‡— t           j        dk    s+t           j        j        dk    st           j        d d…         dvrdS dd l} |                      d¦  «        dk    rdS dd	lm}m	} 	  |d
¦  «        Šn# |$ r Y dS w xY w‰dvrdS dd l
Šdd lŠdd lŠdd l}ddlmŠ	 ddlm} t           j        d d…         dk    Š ‰	t$          ¦  «        j        }‰rdnd}|                     ||z                       d¬¦  «        ¦  «        Š‰r:	  |d¦  «        ‰d         k    s |d¦  «        ‰d         k    rdS n# |$ r Y dS w xY wdd l}‰                     ‰rdnd¦  «        Š‰rdndŠ‰rdndŠ|j        ‰                     d¦  «        d         k    rdS ‰j        Št7          ‰dd¦  «        rdS t9          ‰r‰j        n‰j        d ¦  «        rdS ‰d!         ‰         Š|‰d"         z  Š
d Š |¦   «         Šˆ	ˆ
ˆˆˆˆˆˆˆˆˆˆˆˆfd#„}d|_        |‰_        dS )$NÚwin32Úcpythoné   ))é   é   ©r   é   Fr   ÚPé   )ÚPackageNotFoundErrorÚversionztriton-windows)z3.7.0.post26z3.7.1.post27)ÚPath)ÚLockr	   zmanifest_hip.jsonzmanifest.jsonúutf-8©ÚencodingÚtorchÚtorch_versionzrocm-sdk-coreÚrocm_versionztriton.backends.amd.driverztriton.backends.nvidia.driverÚ	hip_utilsÚ
cuda_utilsÚHIPÚCUDAz.postÚ_wangp_prebuiltTÚinstanceÚbuildsÚbinaryc           	      ó  •— |‰k    r
 ‰| |fi |¤ŽS ‰5  ‰r7 ‰‰j         ¦  «                             d¦  «                             d¬¦  «        n| }‰                     |                     d¦  «        ¦  «                             ¦   «         ‰d         k    rt          d‰› d�¦  «        ‚‰rR‰                     ¦   «                              dd¦  «        }| |                     d	|d
¦  «        k    rt          d¦  «        ‚‰�€q ‰‰j         ¦  «                             d¬¦  «        }‰r‰d         n‰d         }‰                     |                     d¦  «        ¦  «                             ¦   «         |k    rt          d‰› d�¦  «        ‚	 ‰ 	                    ¦   «         }n)# t          $ r}t          d‰› d‰› d�¦  «        |‚d }~ww xY w‰                     |¦  «                             ¦   «         ‰d         k    rt          d‰› d�¦  «        ‚‰j                             ‰‰¦  «        }	‰j                             |	¦  «        }
|	j                             |
¦  «         |
Št!          d‰› d‰› d�¦  «         ‰cd d d ¦  «         S # 1 swxY w Y   d S )Nzdriver.cr   r   Údriver_source_sha256zWanGP's prebuilt zV helper does not match Triton's driver.c. Reinstall the official triton-windows wheel.ú\z\\z/*py_libhip_search_path*/é   z4Unexpected Triton HIP helper source or runtime path.Údriver_python_sha256zW helper does not match Triton's driver.py. Reinstall the official triton-windows wheel.z helper is unavailable: zJ. Restore this file from the WanGP repository and check antivirus history.Úbinary_sha256zG helper checksum differs. Restore the helper from the WanGP repository.z [Triton] Using WanGP's Prebuilt z	 Helper (ú))Ú__file__Ú	with_nameÚ	read_textÚsha256ÚencodeÚ	hexdigestÚRuntimeErrorÚ_get_path_to_hip_runtime_dylibÚreplaceÚ
read_bytesÚOSErrorÚutilÚspec_from_file_locationÚmodule_from_specÚloaderÚexec_moduleÚprint)ÚsrcÚnameÚkwargsÚsourceÚruntime_pathÚdriver_textÚpython_hashÚdataÚerrorÚspecÚmoduler   r   ÚbuildÚdriverÚhashlibÚhelper_nameÚhipÚ	importlibÚlabelÚloadedÚlockÚmanifestÚoriginalÚreleases              €€€€€€€€€€€€€€úFC:\Users\johnn\OneDrive\Documents\AI\shared\triton_runtime\__init__.pyÚcompile_or_loadz0install_prebuilt_helper.<locals>.compile_or_load9   so  ø€ à�;ÒÐØ�8˜C Ð0Ð0¨Ð0Ð0Ð0Øð 	ð 	Ø`cÐl�T�T˜&œ/Ñ*Ô*×4Ò4°ZÑ@Ô@×JÒJÐT[ÐJÑ\Ô\Ð\ÐilˆFØ�~Š~˜fŸmšm¨GÑ4Ô4Ñ5Ô5×?Ò?ÑAÔAÀUÐKaÔEbÒbÐbÝ"ð  $U°uð  $Uð  $Uð  $Uñ  Vô  Vð  VØð _Ø%×DÒDÑFÔF×NÒNÈtÐU[Ñ\Ô\�Ø˜&Ÿ.š.Ð)DÀlÐTUÑVÔVÒVÐVÝ&Ð']Ñ^Ô^Ð^Ø‰~Ø"˜d 6¤?Ñ3Ô3×=Ò=ÀwÐ=ÑOÔO�Ø?BÐh˜eÐ$:Ô;Ð;ÈÐQgÔHh�Ø—>’> +×"4Ò"4°WÑ"=Ô"=Ñ>Ô>×HÒHÑJÔJÈkÒYÐYÝ&ð  (Z¸5ð  (Zð  (Zð  (Zñ  [ô  [ð  [ðyØ!×,Ò,Ñ.Ô.�D�DøÝð yð yð yÝ&ð  (m¸5ð  (mð  (mÐZ`ð  (mð  (mð  (mñ  nô  nð  tyð  yøøøøðyøøøà—>’> $Ñ'Ô'×1Ò1Ñ3Ô3°xÀÔ7PÒPÐPÝ&ð  (J¸5ð  (Jð  (Jð  (Jñ  Kô  Kð  KØ ”~×=Ò=¸kÈ6ÑRÔR�Ø"œ×8Ò8¸Ñ>Ô>�Ø”×'Ò'¨Ñ/Ô/Ð/Ø�ÝÐS¸ÐSÐSÈÐSÐSÐSÑTÔTÐTØð1	ð 	ð 	ð 	ñ 	ô 	ð 	ð 	ð 	ð 	ð 	ð 	øøøð 	ð 	ð 	ð 	ð 	ð 	s7   ”E1I=ÆFÆI=Æ
GÆ%F<Æ<GÇB/I=É=JÊJ) ÚsysÚplatformÚimplementationr8   Úversion_infoÚstructÚcalcsizeÚimportlib.metadatar   r   rD   rG   Úimportlib.utilÚjsonÚpathlibr   Ú	threadingr   r&   ÚparentÚloadsr(   ÚtritonÚimport_moduleÚ__version__ÚsplitÚcompile_module_from_srcÚgetattrÚhasattrÚHIPUtilsÚ	CudaUtilsr   )rT   r   r   rX   r   Ú	directoryÚmanifest_namer]   rO   r   r   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   s            @@@@@@@@@@@@@@rN   Úinstall_prebuilt_helperrh      s>  øøøøøøøøøøøøøø€ å
„|�wÒÐ¥#Ô"4Ô"9¸YÒ"FÐ"FÍ#ÔJZÐ[]Ð\]Ð[]ÔJ^ÐfxÐJxÐJxØˆuØ€M€M€MØ‡‚�sÑÔ˜qÒ Ð ØˆuØ@Ð@Ð@Ð@Ð@Ð@Ð@Ð@ðØ�'Ð*Ñ+Ô+ˆˆøØð ð ð ØˆuˆuðøøøàÐ6Ð6Ð6Øˆuà€N€N€NØÐÐÐØÐÐÐØ€K€K€KØÐÐÐÐÐØÐÐÐÐÐÝ
Ô
˜2˜A˜2Ô
 'Ò
)€CØ�•X‘”Ô%€IØ+.ÐCÐ'Ð'°O€MØ�zŠz˜9 }Ñ4×?Ò?ÈÐ?ÑQÔQÑRÔR€HØ
ð ð	Øˆw�wÑÔ 8¨OÔ#<Ò<Ð<ÀÀÈÑ@XÔ@XÐ\dÐesÔ\tÒ@tÐ@tØ�uð Auøà#ð 	ð 	ð 	Ø�5�5ð	øøøà€M€M€MØ×$Ò$ÀSÐ%mÐ%AÐ%AÐNmÑnÔn€FØ!$Ð6�+�+¨,€KØÐ$ˆEˆE˜f€EàÔ˜WŸ]š]¨7Ñ3Ô3°AÔ6Ò6Ð6ØˆuØÔ-€HÝˆxÐ*¨EÑ2Ô2ð Øˆtõ  #Ð;ˆvŒˆ¨6Ô+;¸ZÑHÔHð Øˆuà�XÔ˜wÔ'€EØ˜ (Ô+Ñ+€FØ€FØˆ4‰6Œ6€Dðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð< '+€OÔ#Ø%4€FÔ"Øˆ4s$   Á4B  Â B	ÂB	Ä*D= Ä=EÅE)Ú__doc__rP   rh   © ó    rN   ú<module>rl      s2   ðà PÐ PØ 
€
€
€
ðSð Sð Sð Sð Srk   