U Mâh µã!@s"UdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl m Z mZmZmZddlmZmZdd lmZGd d „d eƒZGd d „d ƒZe eefZdZiZeed<dddœZddddddddddddddd d!d"d#d$d%d&d'd'd(d)d*d+d,d-d.d d d/g!ZGd0d„deƒZ d1d„Z!Gd2d„dƒZ"Gd3d„dƒZ#eee$d4œd5d„Z%d6d7„Z&dieeeee$e$eed:œd;d„Z'eeeee eefe$e$e#e$dd<œ d=d„Z(d>d„Z)d?d„Z*e +d@¡Z,dAd„Z-dBd„Z.dCd„Z/dDd„Z0e +dE¡Z1dFd„Z2e +dG¡Z3dHd „Z4e +dI¡Z5dJd!„Z6djdKd"„Z7dLd#„Z8dMd$„Z9dNd%„Z:dOd&„Z;dPd'„Ze>ƒZ?iZ@e>ƒZAiZBe eeCfedT<iZDe D]ªZEeFeEeƒs¬tG‚eE H¡D]Š\ZIZJeJdZKeJdd…ZLejMeLkreA NeI¡ejOeLkr eB PeIdU¡r eKeDeI<neKeBeI<ejQeLkr´ejOeLkr´e? NeI¡eKe@eI<q´q˜e +e? R¡¡ZSe +dVeA R¡›dW¡ZTe +dX¡ZUe +dY¡ZVe +dZ¡ZWe +d[¡ZXeeeee eefe$e$e#e$ed<œ d\d)„ZYdkd]d*„ZZd^d+„Z[d_d,„Z\e +d`¡Z]dad-„Z^dbd.„Z_dlee$eeeeeee$ee$e$e$e$ee#edgœdhd/„Z`dS)ma The Python Hipify script. ## # Copyright (c) 2015-2016 Advanced Micro Devices, Inc. All rights reserved. # 2017-2018 Advanced Micro Devices, Inc. and # Facebook Inc. All rights reserved. # # Permission is hereby granted, free of charge, to any person obtaining a copy # of this software and associated documentation files (the "Software"), to deal # in the Software without restriction, including without limitation the rights # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell # copies of the Software, and to permit persons to whom the Software is # furnished to do so, subject to the following conditions: # # The above copyright notice and this permission notice shall be included in # all copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN # THE SOFTWARE. éNé)Ú constants)ÚCUDA_TO_HIP_MAPPINGS)ÚMATH_TRANSPILATIONS)ÚDictÚListÚIteratorÚOptional)ÚMappingÚIterable)ÚEnumc@seZdZdZdZdS)Ú CurrentStateréN)Ú__name__Ú __module__Ú __qualname__Ú INITIALIZEDÚDONE©rrúR/var/www/html/venv/lib/python3.8/site-packages/torch/utils/hipify/hipify_python.pyr *sr c@seZdZdd„Zdd„ZdS)Ú HipifyResultcCs||_||_d|_dS)NÚ©Ú current_stateÚ hipified_pathÚstatus)ÚselfrrrrrÚ__init__/szHipifyResult.__init__cCsd|j›d|j›d|j›S)NzHipifyResult:: current_state: z, hipified_path : z , status: r©rrrrÚ__str__4szHipifyResult.__str__N)rrrrrrrrrr.srz;// !!! This is a file automatically generated by hipify!!! ÚHIPIFY_FINAL_RESULTZscalar_t)ZDtypeÚTÚ InputErrorÚopenfÚbcolorsÚGeneratedFileCleanerÚmatch_extensionsÚmatched_files_iterÚpreprocess_file_and_save_resultÚ compute_statsÚadd_dim3ÚprocessKernelLaunchesÚfind_closure_groupÚfind_bracket_groupÚfind_parentheses_groupÚreplace_math_functionsÚhip_header_magicÚreplace_extern_sharedÚget_hip_file_pathÚis_out_of_placeÚis_pytorch_fileÚis_cusparse_fileÚis_special_fileÚis_caffe2_gpu_fileÚTrieÚ preprocessorÚfile_specific_replacementÚfile_add_headerÚfix_static_global_kernelsÚextract_argumentsÚstr2boolÚhipifycs$eZdZ‡fdd„Zdd„Z‡ZS)r"cstƒ |¡||_dS©N)ÚsuperrÚmessage)rrB©Ú __class__rrrKs zInputError.__init__cCs d|j›S)Nz Input error: )rBrrrrrOszInputError.__str__)rrrrrÚ __classcell__rrrCrr"Hs cCst||ddS)NÚignore)Úerrors)Úopen)ÚfilenameÚmoderrrr#Ssc@s,eZdZdZdZdZdZdZdZdZ dZ d S) r$zzzzzzzzN) rrrÚHEADERZOKBLUEÚOKGREENÚWARNINGZFAILÚENDCZBOLDZ UNDERLINErrrrr$Xsc@s<eZdZdZddd„Zdd„Zdd„Zdd d „Zd d „Zd S)r%z+Context Manager to clean up generated filesFcCs||_tƒ|_g|_dSr@)Úkeep_intermediatesÚsetÚfiles_to_cleanÚ dirs_to_clean)rrOrrrrmszGeneratedFileCleaner.__init__cCs|Sr@rrrrrÚ __enter__rszGeneratedFileCleaner.__enter__cOs0tj |¡s |j tj |¡¡t|f|ž|ŽSr@)ÚosÚpathÚexistsrQÚaddÚabspathrH)rÚfnÚargsÚkwargsrrrrHus zGeneratedFileCleaner.opencCsxtj |¡\}}|s$tj |¡\}}|rF|rFtj |¡sF|j|ddtj |¡rV|stt |¡|j tj  |¡¡dS)NT)Úexist_ok) rTrUÚsplitrVÚmakedirsÚisdirÚmkdirrRÚappendrX)rÚdnr\ÚparentÚnrrrr^zs zGeneratedFileCleaner.makedirscCs@|js<|jD]}t |¡q |jddd…D]}t |¡q,dS)Néÿÿÿÿ)rOrQrTÚunlinkrRÚrmdir)rÚtypeÚvalueÚ tracebackÚfÚdrrrÚ__exit__„s   zGeneratedFileCleaner.__exit__N)F)F) rrrÚ__doc__rrSrHr^rmrrrrr%ks   )rIÚ extensionsÚreturncst‡fdd„|DƒƒS)zŽsz#match_extensions..©Úany)rIrorrurr&Œscst‡fdd„|DƒƒS)Nc3s|]}t ˆ|¡VqdSr@)Úfnmatch)rsÚpattern©Úfilepathrrrv’sz_fnmatch..rw)r|Úpatternsrr{rÚ_fnmatch‘sr~rF)Ú root_pathÚincludesÚignoresroÚout_of_place_onlyÚis_pytorch_extensionrpccsòt|ƒ}tj|ddD]Ö\}}} tj ||¡} | dkrvd|krH| d¡d|krZ| d¡d|krv| d¡| d¡| D]p} tj || ¡} tj | | ¡} t| |ƒrzt| |ƒszt | |ƒsÀ| |krz|sät | ƒsÖt | ƒsÖqz|rät | ƒsäqz| VqzqdS)NT)ÚtopdownÚ.z.gitÚbuildZ third_partyzthird_party/nvfuser) rPrTÚwalkrUÚrelpathÚremoveraÚjoinr~r&r4r7r3)rr€rror‚rƒÚ exact_matchesZ abs_dirpathÚdirsÚ filenamesZ rel_dirpathrIr|Ú rel_filepathrrrr'•s8    ÿþýý ) Úoutput_directoryr|Ú all_filesÚheader_include_dirsÚstatsÚhip_clang_launchrƒÚ clean_ctxÚ show_progressrpc Csttj tj ||¡¡} ttj| d} | t| <t|||||||||ƒ } |rhd| j krht | d| j | j dd| t| <dS)N)rrZignoredz->T)Úflush) rTrUrXrŠrr rr r9rÚprintr) rr|rr‘r’r“rƒr”r•Úfin_pathÚ hipify_resultÚresultrrrr(Ás$  ÿþcCsPdd„|dDƒ}tdt|ƒd›ƒtd |¡ƒtdt|dƒd›ƒdS) NcSsh|] \}}|’qSrr)rsZ cuda_callZ _filepathrrrÚ Ûsz compute_stats..Úunsupported_callsz1Total number of unsupported CUDA function calls: rlú, z+ Total number of replaced kernel launches: Úkernel_launches)r—ÚlenrŠ)r’rœrrrr)ÚscCs¦d}d}| dd¡ dd¡}dd„tdƒDƒ}d||d<t|ƒD]Š\}}|d krVqÎ|d krh|d 7}n|d krx|d 8}|d ks|t|ƒd krB|dkrB||d k||d <|d 7}|dkrB|d ||d<qB||dd|dd d …}||d d|d d …}||dd|dd … dd¡ d¡} ||d d|d d … dd¡ d¡} d| ›d } d| ›d } | | | ¡} | | | ¡}| ||| |¡}|S)zBadds dim3() to the second and third arguments in the kernel launchrú<<>>cSsg|]}i‘qSrr)rsÚ_rrrÚ ìszadd_dim3..rÚstartrú(ú)ú,ÚendÚ ú zdim3()ÚreplaceÚrangeÚ enumeraterŸÚstrip)Ú kernel_stringÚ cuda_kernelÚcountÚclosureZarg_locsÚindÚcZ first_arg_rawZsecond_arg_rawZfirst_arg_cleanZsecond_arg_cleanZfirst_arg_dim3Zsecond_arg_dim3Zfirst_arg_raw_dim3Zsecond_arg_raw_dim3rrrr*çs6    **    z([ ]+)(detail?)::[ ]+\\\n[ ]+csVt dd„ˆ¡‰‡fdd„}dd„}dd„}t||ˆƒƒƒ}ˆ}|D] }||ƒ}ˆ d |d ¡} ˆ|d d | d …} ˆ|d |d …} |d d dkr¢d nd } ˆ|d d || d d …} t| | ƒ}ttd |d dd ¡ dd¡ƒƒ}d|d d… ddd|d¡ dd¡ dd¡ | d | d¡}| | |¡}|d |¡qD|S)zK Replace the CUDA style Kernel launches with the HIP style kernel launches.cSs| d¡›| d¡›dS)Nrrz::©Úgroup©ZinprrrÚóz'processKernelLaunches..c s„|d|ddœdddœdddœdœ}ddi}d}d}d }d }|}t|d ddddƒD]"}ˆ|} |||fkrà| d kr¤||kr”|}||d d<|dd7<| dkrà|dd8<|ddkrà||krà||d d<|}||krZˆ| ¡sˆ|dkrP||kr|}||dd<|dkr~d|dd<|d|d |d gSqZ||krZ||dd<|d|d |d gSqZdS)Nr¤r¨©r¤r¨re)Ú kernel_launchÚ kernel_nameÚtemplatez<>rrrér»ú>r½ú<>ú#r¢r¦r¥ú:r¼)r¬Úisalnum) Z in_kernelÚposr±ZSTARTZ AT_TEMPLATEZAFTER_TEMPLATEZAT_KERNEL_NAMErÚiÚchar©ÚstringrrÚgrab_method_and_templatesDý        z7processKernelLaunches..grab_method_and_templatecSsdd}g}| d|¡dkr`| d|¡}| d|¡d}|dkrDtdƒ‚| |||||…dœ¡q|S)zKFinds the starting and ending points for all kernel launches in the string.rr rer¡r¾zno kernel end found)r¤r¨r¶)Úfindr"ra)rÈZ kernel_endZkernel_positionsZ kernel_startrrrÚfind_kernel_boundsMs  ÿ z1processKernelLaunches..find_kernel_boundscSsâd}d}d}|D]Ì}|dkrf|dkr2|dkr2d}q¾|dkrH|dkrHd}q¾|dkr¾|dkr¾|dkr¾d}nX|dkr„|d ks~|d kr¾d}n:|dkr¢|dkr¾|dkr¾d}n|dkr¾|dkr¾|dkr¾d}|}|dkrÔ||7}q|d 7}q|S) Nrú/z//Ú*z/*ú"ú\ú'ú r©Úxr)rÈZ in_commentZprev_cZ new_stringr´rrrÚ mask_commentses2  z,processKernelLaunches..mask_commentsr¥r¨rr¤rrer¶r r¡r¦zhipLaunchKernelGGL(z, 0érrž) ÚRE_KERNEL_LAUNCHÚsubÚlistrÊr*rŸr=r«ra)rÈr’rÉrËrÓZget_kernel_positionsÚ output_stringZkernelÚparamsZ parenthesisr°r¯Zend_param_indexZkernel_name_with_templateZcuda_kernel_dim3Znum_klpZ hip_kernelrrÇrr+ s> ;!   "ÿÿþ þ cCsŽd}d}|}d\}}|t|ƒkrŠ|||dkrP|dkrFd}d}|}q€|d7}n0|||dkr€|r€|d8}|dkr€|}||fS|d7}qdS)aÊGeneralization for finding a balancing closure group if group = ["(", ")"], then finds the first balanced parentheses. if group = ["{", "}"], then finds the first balanced bracket. Given an input string, a starting position in the input string, and the group type, find_closure_group returns the positions of group[0] and group[1] as a tuple. Example: >>> find_closure_group("(hi)", 0, ["(", ")"]) (0, 3) Fr)rereTr)NN)rŸ)Ú input_stringr¤r¶Zinside_parenthesisZparensrÄZp_startZp_endrrrr,§s$   cCst||ddgdS)z%Finds the first balanced parantheses.Ú{Ú}rµ©r,©rÚr¤rrrr-ÍscCst||ddgdS)z!Finds the first balanced bracket.r¥r¦rµrÝrÞrrrr.Òsz\bassert[ ]*\(cCs.|}tD] }| |›dt|›d¡}q|S)a‹FIXME: Temporarily replace std:: invocations of math functions with non-std:: versions to prevent linker errors NOTE: This can lead to correctness issues when running tests, since the correct version of the math function (exp/expf) might not get called. Plan is to remove this function once HIP supports std:: math function calls inside device code r¥)rr«)rÚrØÚfuncrrrr/Ús z:?:?\b(__syncthreads)\b(\w*\()csh|‰ddg}t‡fdd„|Dƒƒr&ˆSdˆk}|dˆk7}|dˆk7}|t ˆ¡dk 7}|rdd |‰ˆS) aIf the file makes kernel builtin calls and does not include the cuda_runtime.h header, then automatically add an #include to match the "magic" includes provided by NVCC. TODO: Update logic to ignore cases where the cuda_runtime.h is included by another file. zhip/hip_runtime.hzhip/hip_runtime_api.hc3s(|] }t d|›d|›dˆ¡VqdS)z #include ("z"|)N)ÚreÚsearch)rsÚext©rØrrrvùsz#hip_header_magic..ZhipLaunchKernelGGLÚ __global__Z __shared__Nz#include "hip/hip_runtime.h" )rxÚRE_SYNCTHREADSrá)rÚÚheadersZhasDeviceLogicrrãrr0ís  zGextern\s+([\w\(\)]+)?\s*__shared__\s+([\w:<>\s]+)\s+(\w+)\s*\[\s*\]\s*;cCs|}t dd„|¡}|S)aMatch extern __shared__ type foo[]; syntax and use HIP_DYNAMIC_SHARED() MACRO instead. https://github.com/ROCm-Developer-Tools/HIP/blob/master/docs/markdown/hip_kernel_language.md#__shared__ Example: "extern __shared__ char smemChar[];" => "HIP_DYNAMIC_SHARED( char, smemChar)" "extern __shared__ unsigned char smem[];" => "HIP_DYNAMIC_SHARED( unsigned char, my_smem)" cSs.d| d¡pd›d| d¡›d| d¡›dS) NzHIP_DYNAMIC_SHARED(rrrªrrr¾r¦rµr·rrrr¸r¹z'replace_extern_shared..)ÚRE_EXTERN_SHAREDrÖ)rÚrØrrrr1 s ÿcCsðtj |¡rt‚|s t|ƒs |Stj |¡\}}tj |¡\}}|dkrLd}|}|}| dd¡}| dd¡}| dd¡}| dd¡}| dd¡}|d kr¤| dd¡}|s¾||kr¾tj |d¡}|rÞ||krÞ|||krÞ|d }tj |||¡S) z3 Returns the new name of the hipified file ú.cuú.hipÚcudaZhipÚCUDAÚHIPÚTHCÚTHHz caffe2/coreZ_hip) rTrUÚisabsÚAssertionErrorr3r]Úsplitextr«rŠ)rŽrƒÚdirpathrIÚrootrâÚ orig_filenameZ orig_dirpathrrrr2s* $       cCs>tj |¡rt‚| d¡rdS| d¡r,dS| d¡r:dSdS)Nútorch/Fúthird_party/nvfuser/útools/autograd/templates/T©rTrUrïrðÚ startswith©rŽrrrr3cs   cCsZtj |¡rt‚| d¡r,| d¡r(dSdS| d¡r:dS| d¡rHdS| d¡rVdSdS)Nzaten/zaten/src/ATen/core/FTrõrör÷rørúrrrr4os     cCst|ƒrd| ¡kSdS)NÚsparseF©r4Úlowerrúrrrr5~s cCs<t|ƒr8d| ¡krdSd| ¡kr8d| ¡kr4dSdSdS)NrûTZlinalgZbatchlinearalgebralibblasFrürúrrrr6„s   cCsRtj |¡rt‚| d¡rdStj |¡}tj |¡\}}d|ksJ|dkoPd|kS)Nzc10/cudaTZgpu©rèú.cuhZcudnn)rTrUrïrðrùÚbasenamerñ)rŽrIr¢rârrrr7Žs   c@seZdZdZdd„ZdS)ÚTrieNodezA Trie node whose children are represented as a directory of char: TrieNode. A special char '' represents end of word cCs i|_dSr@)Úchildrenrrrrr›szTrieNode.__init__N)rrrrnrrrrrr–src@sPeZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dd„Z dd„Z dS)r8z£Creates a Trie out of a list of words. The trie can be exported to a Regex pattern. The corresponding Regex should match much faster than a simple Regex union.cCs tƒ|_dS)z,Initialize the trie with an empty root node.N)rrórrrrr¢sz Trie.__init__cCs8|j}|D]}|j |tƒ¡|j|}q d|jd<dS)zAdd a word to the Trie. TrN)rórÚ setdefaultr©rÚwordÚnoderÆrrrrW¦s  zTrie.addcCs|jS)zReturn the root node of Trie. )rórrrrÚdump¯sz Trie.dumpcCs t |¡S)z Escape a char for regex. )ràÚescape)rrÆrrrÚquote³sz Trie.quotecCs6|j}|D] }||jkr$|j|}q dSq d|jkS)zZSearch whether word is present in the Trie. Returns True if yes, else return FalseFr)rórrrrrrá·s   z Trie.searchc CsF|}d|jkr$t|j ¡ƒdkr$dSg}g}d}t|j ¡ƒD]h}t|j|tƒr¢z(| |j|¡}| | |¡|¡Wq¦t k rž| | |¡¡Yq¦Xq>d}q>t|ƒdk }t|ƒdkröt|ƒdkrÞ| |d¡n| dd  |¡d¡t|ƒdkr|d} ndd  |¡d } |rB|r6| d 7} n d| ›d } | S) z0Convert a Trie into a regular expression patternrrNrú[ú]z(?:ú|r¦ú?z)?) rrŸÚkeysÚsortedÚ isinstancerÚ_patternrar Ú ExceptionrŠ) rrórZaltÚccÚqrÆÚrecurseZcconlyršrrrrÄs6     z Trie._patterncCs | |j¡S©z#Export the Trie to a regex pattern.©rrórrrrrzësz Trie.patterncCs | |j¡SrrrrrrÚexport_to_regexïszTrie.export_to_regexN) rrrrnrrWrr rárrzrrrrrr8žs  'Ú PYTORCH_MAPrz(?<=\W)(z)(?=\W)z#include "([^"]+)"z#include <([^>]+)>z"#define THC_GENERIC_FILE "([^"]+)"z\.cu\bc  s|tj tj ˆ|¡¡‰tˆ} |ˆkr>d| _d| _tj| _ | Stj  |ˆ¡} t ˆddH} |   ¡t krŠd| _d| _tj| _ | W5QR£S|  d¡|  ¡} W5QRX| } tj tj ˆt| ˆƒ¡¡}tj tj |¡¡s술tj |¡¡dd„‰‡fd d „}ˆrt ˆ| ¡} nDt| ƒr,t || ¡} n,t| ƒrDt ˆ| ¡} nd d „}t || ¡} d'‡‡‡‡‡‡‡‡‡ f dd„ }t |dd ƒ| ¡} t |ddƒ| ¡} t |dƒ| ¡} | d¡rÚ|  dd¡} |  dd¡} t d| ¡} ˆsêt| ˆ ƒ} | d¡rd|krt | ƒ} t!| ƒ} ˆrR| | krRtj ˆ¡tj |¡krRˆ| _d| _tj| _ | Sˆ|krpt"ˆdƒrpt | } d }tj |¡r¦t |dd}| ¡| k}W5QRX|r`z@ˆj |ddd}| #| ¡W5QRX|| _d| _tj| _ | WSt$k r\}zPt%t&j'›d |›d!|j(›d"ˆ›d#t&j)› t*j+d$ˆ| _d%| _tj| _ | WY¢Sd}~XYnXn|| _d&| _tj| _ | SdS)(z< Executes the CUDA -> HIP conversion on the specified file. Nz[ignored, not to be hipified]zutf-8)Úencodingz#[ignored, input is hipified output]rcSst| d¡S©Nr)rr¶©ÚmrrrÚpt_replKszpreprocessor..pt_replcst | d¡ˆ|ƒ¡Sr)ÚPYTORCH_SPECIAL_MAPÚgetr¶r)rrrÚpt_special_replNsz%preprocessor..pt_special_replcSst| d¡Sr)Ú CAFFE2_MAPr¶rrrrÚc2_repl[szpreprocessor..c2_replTc s$‡‡‡‡‡‡‡‡‡ ‡ ‡f dd„}|S)Nc sà| d¡}tj |¡\}‰| d¡s8| d¡rN| d¡sNˆ  t| d¡ˆƒ¡SˆrÖt‡fdd„ˆDƒƒrÖd}d}ˆr®tj ˆ¡}tj  tj  ||¡¡}tj  |¡r®|}|}|dkrøˆD]<}tj  ˆ|¡}tj  tj  ||¡¡}tj  |¡rº|}|}qº|dkr | d¡S|t kr0t ˆ|ˆˆˆ ˆˆˆˆ ƒ nz|t krªt |}|jtjkrªtj |ˆ¡} tj  tj  ˆt| ˆƒ¡¡} | |_|t |<ˆ  tj | dk r | n||¡¡St |j} ˆ  tj | dk rÌ| n||¡¡S| d¡S)Nr)z ATen/cudazATen/native/cudazATen/native/nested/cudazATen/native/quantized/cudazATen/native/sparse/cudazATen/native/transformers/cudazTHC/ríZTHCPc3s|]}| ˆ¡VqdSr@rq)rsÚsrurrrvpsz>preprocessor..mk_repl..repl..r)r¶rTrUr]rùÚformatr2rxÚdirnamerXrŠrVr r(rr rrˆr) rrkròZ header_dirZheader_filepathZheader_dir_to_checkZheader_path_to_checkÚheader_include_dirZ header_resultZheader_rel_pathZheader_fout_pathZhipified_header_filepath) rr”r˜r‘r“Úinclude_current_dirrƒrr•r’ÚtemplrurÚreplasr ÿøø       ý ÿÿ ÿz+preprocessor..mk_repl..replr)r)r(r*) rr”r˜r‘r“rƒrr•r’)r(r)rÚmk_repl`s :zpreprocessor..mk_replz#include "{0}"z#include <{0}>Fz#define THC_GENERIC_FILE "{0}"zCMakeLists.txtrërìrírîrérþZ PowKernelz[skipped, no changes])rèrÿú.cú.ccú.cppú.hú.hppÚwz[ok]zFailed to save z with "z ", leaving z unchanged.©Úfilez[skipped, no permissions]zkipped, already hipified])T),rTrUrXrŠr rrr rrrˆrHÚreadlineÚHIPIFY_C_BREADCRUMBÚseekÚreadr2rVr&r^ÚRE_PYTORCH_PREPROCESSORrÖr6r4ÚRE_CAFFE2_PREPROCESSORÚRE_QUOTE_HEADERÚRE_ANGLE_HEADERÚRE_THC_GENERIC_FILErrr«Ú RE_CU_SUFFIXr+r/r0r&ÚwriteÚPermissionErrorr—r$rMÚstrerrorrNÚsysÚstderr)rr|rr‘r’r“rƒr”r•r™rŽZfinZ output_sourceZorig_output_sourceZ fout_pathr!r#r+Zdo_writeZfout_oldZfoutrtr) rr”r˜r‘r“rƒrrr•r’rr9%sž       <     ÿþý&ÿc stt|dƒ`}| ¡}|r>t dt |¡›d‡fdd„|¡}n | |ˆ¡}| d¡| |¡| ¡W5QRXdS)Núr+z\b(z)\bcsˆSr@r)rÒ©Úreplace_stringrrr¸ár¹z+file_specific_replacement..r) r#r7ràrÖrr«r6r>Útruncate)r|Z search_stringrEÚstrictrkÚcontentsrrDrr:Ýs &   c Csrt|dƒ^}| ¡}|ddkr8|ddkr8d|›d}d|›d|}| d¡| |¡| ¡W5QRXdS) NrCrrÀrer¿rÎz #include z )r#r7r6r>rF)r|ÚheaderrkrHrrrr;és    cCs| dd¡}|S)z+s   ©rèrÿr,r-r.r/z.inr0©rÿr/r0©rÍT)Úproject_directoryÚ show_detailedroÚheader_extensionsrr‘r€Ú extra_filesr‚rr•r“rƒÚhipify_extra_files_onlyr”rpc sòˆdkrt ¡‰tj ˆ¡s.tdƒt d¡ˆsDˆ d¡ˆd‰ˆˆkrt‡‡fdd„|Dƒ}‡‡fdd„| Dƒ} tj ˆ¡sŒt  ˆˆ¡t t ˆ|| ||| d ƒ}t |ƒ}|D]0}tj  |¡sÎtj ˆ|¡}||kr°| |¡q°d d lm}|D]†}tj  |¡r||ƒ}n|tj ˆ|¡ƒ}| d ¡D]L}| ¡r*tt|ƒ|ƒr*tt|ƒ| ƒs*t|j|ƒr*| t|ƒ¡q*qò|dkrŽtd d}ggdœ}| s¢|n|D]}tˆ||||| | || ƒ q¦ttjdtjtjd|rît|ƒtS)Nrz,The project folder specified does not exist.rrÌZ_amdcsg|]}| ˆˆ¡‘qSrrJ)rsÚinclude©rr\rrr£Uszhipify..csg|]}| ˆˆ¡‘qSrrJ)rsrFrbrrr£Vs)r€rror‚rƒr)ÚPathrÍT)rO)rœržz-Successfully preprocessed all matching files.r2) rTÚgetcwdrUrVr—rAÚexitÚrstripÚshutilÚcopytreer×r'rPrïrŠraÚpathlibrcÚrglobÚis_filer~Ústrr&Únamer%r(r$rLrNrBr)r )r\r]ror^rr‘r€r_r‚rr•r“rƒr`r”rZ all_files_setrkrcr'Zheader_include_dir_pathrUr’r|rrbrr?6sn     ý    ÿ þ ý ü    ÿ)rrrFF)F)F)FrYrZrrr[rFrTFFFN)arnrVryràrgrArTrrZcuda_to_hip_mappingsrrÚtypingrrrr Úcollections.abcr r Úenumr r rrlZHipifyFinalResultr5r Ú__annotations__ZPYTORCH_TEMPLATE_MAPÚ__all__rr"r#r$r%Úboolr&r~r'r(r)r*ÚcompilerÕr+r,r-r.Z RE_ASSERTr/rår0rçr1r2r3r4r5r6r7rr8Z CAFFE2_TRIEr"Z PYTORCH_TRIErÚobjectrÚmappingrrðÚitemsÚsrcriÚdstZ meta_dataZ API_CAFFE2rWZ API_SPECIALr Z API_PYTORCHrr9r8r:r;r<r=r9r:r;r<Z RE_INCLUDEr=r>r?rrrrÚs\        û !úú - ÷  # &    H  U           ÷ 9   . ñð