U ‰d¢ã@sžUdZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl m Z mZmZmZddlmZmZe eeefZe eefZd ZiZeed <d d d œZGd d„deƒZdd„ZGdd„dƒZGdd„dƒZeee dœdd„Z!dd„Z"d_eeeee e eedœdd„Z#eeeee eefe e ee ddœ d d!„Z$d"d#„Z%d$d%„Z&e 'd&¡Z(d'd(„Z)d)d*„Z*d+d,„Z+d-d.„Z,e 'd/¡Z-d0d1„Z.e 'd2¡Z/d3d4„Z0e 'd5¡Z1d6d7„Z2d`d8d9„Z3d:d;„Z4dd?„Z6d@dA„Z7GdBdC„dCƒZ8e8ƒZ9iZ:e8ƒZ;iZe D]žZ?e@e?eƒs6tA‚e? B¡D]~\ZCZDeDdZEeDdd…ZFejGeFkrœe; HeC¡ejIeFkr”e< JeCdE¡r”eEe>eC<neEee9 HeC¡eEe:eC<q>q"e 'e9 L¡¡ZMe 'dF Ne; L¡¡¡ZOe 'dG¡ZPe 'dH¡ZQe 'dI¡ZRe 'dJ¡ZSeeeee eefe e ee edœ dKdL„ZTdadMdN„ZUdOdP„ZVdQdR„ZWe 'dS¡ZXdTdU„ZYdVdW„ZZdbee eeeeeee ee e e e eeed\œd]d^„Z[dS)ca 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ÚIterablez;// !!! This is a file automatically generated by hipify!!! ÚHIPIFY_FINAL_RESULTZscalar_t)ZDtypeÚTcs$eZdZ‡fdd„Zdd„Z‡ZS)Ú InputErrorcstt|ƒ |¡||_dS©N)ÚsuperrÚ__init__Úmessage)Úselfr©Ú __class__©úD/tmp/pip-unpacked-wheel-ua33x9lu/torch/utils/hipify/hipify_python.pyr5szInputError.__init__cCsd d|j¡S)Nz{}: {}z Input error)Úformatr©rrrrÚ__str__9szInputError.__str__)Ú__name__Ú __module__Ú __qualname__rrÚ __classcell__rrrrr2s rcCst||ddS)NÚignore)Úerrors)Úopen)ÚfilenameÚmoderrrÚopenf=sr$c@s,eZdZdZdZdZdZdZdZdZ dZ d S) ÚbcolorszzzzzzzzN) rrrÚHEADERZOKBLUEÚOKGREENÚWARNINGÚFAILÚENDCZBOLDZ UNDERLINErrrrr%Bsr%c@s<eZdZdZddd„Zdd„Zdd„Zdd d „Zd d „Zd S)ÚGeneratedFileCleanerz+Context Manager to clean up generated filesFcCs||_tƒ|_g|_dSr)Úkeep_intermediatesÚsetÚfiles_to_cleanÚ dirs_to_clean)rr,rrrrVszGeneratedFileCleaner.__init__cCs|SrrrrrrÚ __enter__[szGeneratedFileCleaner.__enter__cOs0tj |¡s |j tj |¡¡t|f|ž|ŽSr)ÚosÚpathÚexistsr.ÚaddÚabspathr!)rÚfnÚargsÚkwargsrrrr!^s zGeneratedFileCleaner.opencCsxtj |¡\}}|s$tj |¡\}}|rF|rFtj |¡sF|j|ddtj |¡rV|stt |¡|j tj  |¡¡dS)NT)Úexist_ok) r1r2Úsplitr3ÚmakedirsÚisdirÚmkdirr/Úappendr5)rÚdnr9ÚparentÚnrrrr;cs zGeneratedFileCleaner.makedirscCs@|js<|jD]}t |¡q |jddd…D]}t |¡q,dS)Néÿÿÿÿ)r,r.r1Úunlinkr/Úrmdir)rÚtypeÚvalueÚ tracebackÚfÚdrrrÚ__exit__ms   zGeneratedFileCleaner.__exit__N)F)F) rrrÚ__doc__rr0r!r;rJrrrrr+Ts   r+)r"Ú extensionsÚreturncst‡fdd„|DƒƒS)zvsz#match_extensions..©Úany)r"rLrrRrÚmatch_extensionstsrVcst‡fdd„|DƒƒS)Nc3s|]}t ˆ|¡VqdSr)Úfnmatch)rPÚpattern©ÚfilepathrrrSysz_fnmatch..rT)rZÚpatternsrrYrÚ_fnmatchxsr\rF)Ú root_pathÚincludesÚignoresrLÚout_of_place_onlyÚis_pytorch_extensionrMccsèt|ƒ}tj|ddD]Ì\}}} tj ||¡} | dkrld|krH| d¡d|krZ| d¡d|krl| d¡| D]p} tj || ¡} tj | | ¡} t| |ƒrpt| |ƒspt| |ƒs¶| |krp|sÚt | ƒsÌt | ƒsÌqp|rÚt | ƒsÚqp| VqpqdS)NT)ÚtopdownÚ.z.gitÚbuildZ third_party) r-r1Úwalkr2ÚrelpathÚremoveÚjoinr\rVÚis_pytorch_fileÚis_caffe2_gpu_fileÚis_out_of_place)r]r^r_rLr`raZ exact_matchesZ abs_dirpathÚdirsÚ filenamesZ rel_dirpathr"rZÚ rel_filepathrrrÚmatched_files_iter{s6   ÿþýý ro) Úoutput_directoryrZÚ all_filesÚheader_include_dirsÚstatsÚhip_clang_launchraÚ clean_ctxÚ show_progressrMc Csht|||||||||ƒ } tj tj ||¡¡} |r\dt| dƒkr\t| d| d| ddd| t| <dS)NZignoredÚstatusz->Ú hipified_pathT)Úflush)Ú preprocessorr1r2r5rhÚstrÚprintr ) rprZrqrrrsrtrarurvÚresultÚfin_pathrrrÚpreprocess_file_and_save_result¦s ÿþrcCsLdd„|dDƒ}td t|ƒ¡ƒtd |¡ƒtd t|dƒ¡ƒdS)NcSsh|] \}}|’qSrr)rPZ cuda_callZ _filepathrrrÚ ¾sz compute_stats..Úunsupported_callsz6Total number of unsupported CUDA function calls: {0:d}ú, z0 Total number of replaced kernel launches: {0:d}Úkernel_launches)r|rÚlenrh)rsrrrrÚ compute_stats½sr…cCs¢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 | ¡} | | | ¡} | | | ¡}| ||| |¡}|S)zBadds dim3() to the second and third arguments in the kernel launchrú<<<Úú>>>cSsg|]}i‘qSrr)rPÚ_rrrÚ Ïszadd_dim3..éÚstartrú(ú)ú,ÚendÚ ú zdim3({}))ÚreplaceÚrangeÚ enumerater„Ústripr)Ú 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_dim3rrrÚadd_dim3Ês6    **    rz([ ]+)(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.cSsd | d¡| d¡¡S)Nz{0}{1}::rr‹©rÚgroup©ÚinprrrÚóóz'processKernelLaunches..c sˆ|d|ddœdddœdddœdœ}ddi}d}d}d }d }|}t|d ddddƒD]&}ˆ|} ||ksx||krä| d kr¨||kr˜|}||d d<|dd7<| dkrä|dd8<|ddkrä||krä||d d<|}||krZˆ| ¡sˆ|dkrT||kr"|}||dd<|dkr‚d|dd<|d|d |d gSqZ||krZ||dd<|d|d |d gSqZdS)NrŒr©rŒrrB)Ú kernel_launchÚ kernel_nameÚtemplatez<>rrr‹ér¥ú>r§ú<>ú#r‰rú:rŽr¦)r”Úisalnum) Z in_kernelÚposr™ÚSTARTZ AT_TEMPLATEZAFTER_TEMPLATEZAT_KERNEL_NAMErwÚiÚchar©ÚstringrrÚgrab_method_and_templateõsDý       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†rBrˆr¨zno kernel end found)rŒrrŸ)Úfindrr>)r³Z kernel_endZkernel_positionsZ kernel_startrrrÚfind_kernel_bounds0s  ÿ 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_commentsHs2  z,processKernelLaunches..mask_commentsrrrrŒrrBrŸr†rˆrŽzhipLaunchKernelGGL(z, 0ér‚rƒ) ÚRE_KERNEL_LAUNCHÚsubÚlistrµrr„Úextract_argumentsr“r>)r³rsr´r¶r¾Zget_kernel_positionsÚ output_stringÚkernelÚparamsZ parenthesisr˜r—Zend_param_indexZkernel_name_with_templateZcuda_kernel_dim3Znum_klpZ hip_kernelrr²rÚprocessKernelLaunchesðs> ;!   "ÿÿþ þ rÇ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, ["(", ")"]) Returns: 0, 3 Fr)rBrBTr)NN)r„)Ú input_stringrŒrŸZinside_parenthesisÚparensr®Zp_startZp_endrrrÚfind_closure_groupŠs$   rÊcCst||ddgdS)z%Finds the first balanced parantheses.Ú{Ú}©rŸ©rÊ©rÈrŒrrrÚfind_bracket_group²srÐcCst||ddgdS)z!Finds the first balanced bracket.rrŽrÍrÎrÏrrrÚfind_parentheses_group·srÑz\bassert[ ]*\(cCs.|}tD] }| d |¡d t|¡¡}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 z{}()rr“r)rÈrÄÚfuncrrrÚreplace_math_functions¿s rÓ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 |¡ˆ¡VqdS)z#include ("{0}"|<{0}>)N)ÚreÚsearchr)rPÚext©rÄrrrSÞsz#hip_header_magic..ZhipLaunchKernelGGLÚ __global__Z __shared__Nz#include "hip/hip_runtime.h" )rUÚRE_SYNCTHREADSrÕ)rÈÚheadersZhasDeviceLogicrr×rÚhip_header_magicÒs  rÛ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¡¡S)Nz HIP_DYNAMIC_SHARED({0} {1}, {2})rr‡r‹r¨ržr rrrr¢ûs ÿz'replace_extern_shared..)ÚRE_EXTERN_SHAREDrÁ)rÈrÄrrrÚreplace_extern_sharedòs þrÝ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ú.hipZcudaZhipÚCUDAÚHIPÚTHCÚTHHz caffe2/coreZ_hip) r1r2ÚisabsÚAssertionErrorrkr:Úsplitextr“rh)rnraÚdirpathr"ÚrootrÖÚ orig_filenameZ orig_dirpathrrrÚget_hip_file_paths* $       rêcCs0tj |¡rt‚| d¡rdS| d¡r,dSdS)Nútorch/Fútools/autograd/templates/T©r1r2räråÚ startswith©rnrrrrkIs   rkcCsLtj |¡rt‚| d¡r,| d¡r(dSdS| d¡r:dS| d¡rHdSdS)Nzaten/zaten/src/ATen/core/FTrërìrírïrrrriSs    ricCst|ƒrd| ¡kSdS)NÚsparseF)riÚlowerrïrrrÚis_cusparse_file`s ròcCsRtj |¡rt‚| d¡rdStj |¡}tj |¡\}}d|ksJ|dkoPd|kS)Nzc10/cudaTZgpu)rÞú.cuhZcudnn)r1r2rärårîÚbasenameræ)rnr"r‰rÖrrrrjes   rjc@s@eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd d „Z dS)ÚTriezºRegex::Trie in Python. 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 i|_dSr©Údatarrrrrssz Trie.__init__cCs<|j}|D]$}||kr||p i||<||}q d|d<dS)Nrr‡rö)rÚwordÚrefr±rrrr4vs  zTrie.addcCs|jSrrörrrrÚdump}sz Trie.dumpcCs t |¡Sr)rÔÚescape)rr±rrrÚquote€sz Trie.quotec Cs8|}d|kr t| ¡ƒdkr dSg}g}d}t| ¡ƒD]d}t||tƒr˜z&| ||¡}| | |¡|¡Wqœtk r”| | |¡¡YqœXq8d}q8t|ƒdk }t|ƒdkrìt|ƒdkrÔ| |d¡n| dd  |¡d¡t|ƒdkr|d} ndd  |¡d} |r4|r,| d 7} nd | } | S) Nr‡rrú[ú]z(?:ú|rŽú?z(?:%s)?) r„ÚkeysÚsortedÚ isinstanceÚdictÚ_patternr>rüÚ Exceptionrh) rZpDatar÷ZaltÚccÚqr±ZrecurseZcconlyr}rrrrƒs6    z Trie._patterncCs| | ¡¡Sr)rrúrrrrrX¨sz Trie.patternN) rrrrKrr4rúrürrXrrrrrõos%rõÚ PYTORCH_MAPr‡z(?<=\W)({0})(?=\W)z#include "([^"]+)"z#include <([^>]+)>z"#define THC_GENERIC_FILE "([^"]+)"z\.cu\bc  s6|ˆkrdddœStj tj ˆ|¡¡‰tj |ˆ¡} tˆddd:} |  ¡tkrhdddœW5QR£S|  d¡|   ¡} W5QRX| } tj tj ˆt | ˆƒ¡¡} tj  tj  | ¡¡sʈ  tj  | ¡¡d d „‰‡fd d „}ˆrðt ˆ| ¡} nDt| ƒrt || ¡} n,t| ƒr t ˆ| ¡} nd d„}t || ¡} d*‡‡‡‡‡‡‡‡‡ f dd„ }t |ddƒ| ¡} t |ddƒ| ¡} t |dƒ| ¡} | d¡r¶|  dd¡} |  dd¡} t d| ¡} ˆsÆt| ˆ ƒ} | d¡sÞ| d¡rðd|krðt| ƒ} t| ƒ} t| ƒ} ˆr4| | kr4tj  ˆ¡tj  | ¡kr4ˆddœSˆ| krRtˆd ƒrRt| } d}tj  | ¡rŠt| ddd}|  ¡| k}W5QRX|r(z2ˆj| d!dd}| | ¡W5QRX| d"dœWStk r$}zBt t!j"›d#| ›d$|j#›d%ˆ›d&t!j$› t%j&d'ˆd(dœWY¢Sd}~XYnXn | d)dœSdS)+z< Executes the CUDA -> HIP conversion on the specified file. Nz[ignored, not to be hipified])rxrwÚrzutf-8)Úencodingz#[ignored, input is hipified output]rcSst| d¡S©Nr)r rŸ©ÚmrrrÚpt_repløszpreprocessor..pt_replcst | d¡ˆ|ƒ¡Sr )ÚPYTORCH_SPARSE_MAPÚgetrŸr )rrrÚpt_sparse_replûsz$preprocessor..pt_sparse_replcSst| d¡Sr )Ú CAFFE2_MAPrŸr rrrÚc2_replszpreprocessor..c2_replTc s$‡‡‡‡‡‡‡‡‡ ‡ ‡f dd„}|S)Nc s¤| d¡}tj |¡\}‰| d¡st| d¡st| d¡st| d¡st| d¡st| d¡st| d¡st| d ¡rŠ| d ¡sŠˆ  t| d¡ˆƒ¡Sˆršt‡fd d „ˆDƒƒršd}d}ˆrêtj ˆ¡}tj  tj  ||¡¡}tj  |¡rê|}|}|dkr6ˆD]<}tj  ˆ|¡}tj  tj  ||¡¡}tj  |¡rø|}|}qø|dkrJ| d ¡S|t krlt ˆ|ˆˆˆ ˆˆˆˆ ƒ t |d}ˆ  tj |dk r|n||¡¡S| d ¡S)Nrz ATen/cudazATen/native/cudazATen/native/nested/cudazATen/native/quantized/cudazATen/native/sparse/cudazATen/native/transformers/cudazTHC/râZTHCPc3s|]}| ˆ¡VqdSrrN)rPÚsrRrrrSsz>preprocessor..mk_repl..repl..rrx)rŸr1r2r:rîrrêrUÚdirnamer5rhr3r rrf) rrHrçÚ header_dirZheader_filepathZheader_dir_to_checkZheader_path_to_checkÚheader_include_dirZhipified_header_filepath) rqrur~rrrtÚinclude_current_dirrarprvrsÚtemplrRrÚrepl sp ÿþýüûúùøø        ý ÿz+preprocessor..mk_repl..replr)rrr) rqrur~rrrtrarprvrs)rrrÚmk_repl s /zpreprocessor..mk_replz#include "{0}"z#include <{0}>Fz#define THC_GENERIC_FILE "{0}"zCMakeLists.txtràrá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)'r1r2r5rhrfr!ÚreadlineÚHIPIFY_C_BREADCRUMBÚseekÚreadrêr3rr;ÚRE_PYTORCH_PREPROCESSORrÁròriÚRE_CAFFE2_PREPROCESSORÚRE_QUOTE_HEADERÚRE_ANGLE_HEADERÚRE_THC_GENERIC_FILErOr“Ú RE_CU_SUFFIXrÇrÓrÛrÝrVÚwriteÚPermissionErrorr|r%r(Ústrerrorr*ÚsysÚstderr)rprZrqrrrsrtrarurvrnZfinZ output_sourceZorig_output_sourceZ fout_pathrrrZdo_writeZfout_oldZfoutrQr) rqrur~rrrtrarprrvrsrrzÙsz        1     "ÿþý  &ÿ"rzc srt|dƒ^}| ¡}|r.r) r$r(rÔrÁrrûr“r'r/Útruncate)rZZ search_stringr6ÚstrictrHÚcontentsrr5rÚfile_specific_replacementqs $   r:c Csnt|dƒZ}| ¡}|ddkr6|ddkr6d |¡}d |¡|}| d¡| |¡| ¡W5QRXdS)Nr4rrªrBr©z"{0}"z#include {0} )r$r(rr'r/r7)rZÚheaderrHr9rrrÚfile_add_header}s    r<cCs| dd¡}|S)zz #include .*\ncCs6g}dddœ}|}|d}|t|ƒkr2||dkrF|dd7<nt||dkrd|dd8<nV||dkr‚|dd7<n8||dkrº||ddkrº|ddkrº|dd8<|ddkræ|ddkræ| ||d œ¡q2|ddkr(|ddkr(||d kr(| ||d œ¡|d}|d7}q|S) ad Return the list of arguments in the upcoming function parameter closure. Example: string (input): '(blocks, threads, 0, THCState_getCurrentStream(state))' arguments (output): '[{'start': 1, 'end': 7}, {'start': 8, 'end': 16}, {'start': 17, 'end': 19}, {'start': 20, 'end': 53}]' r)rªrrrrŽrªr©ú-r¤r)r„r>)rŒr³Ú argumentsZclosuresÚcurrent_positionZargument_start_posrrrrÑs. þ   (* rÃcCs.| ¡dkrdS| ¡dkr dSt d¡‚dS)zArgumentParser doesn't support type=bool. Thus, this helper method will convert from possible string types to True / False.)ÚyesÚtrueÚtÚyÚ1T)ÚnoÚfalserHrAÚ0FzBoolean value expected.N)rñÚargparseÚArgumentTypeError)ÚvrrrÚstr2bool¿s   rM©rÞrórrrr z.inr!©rór r!©r¸T)Úproject_directoryÚ show_detailedrLÚheader_extensionsrprrr^Ú extra_filesr`r_rvrtraÚhipify_extra_files_onlyrurMc 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)Nr‡z,The project folder specified does not exist.rr·Z_amdcsg|]}| ˆˆ¡‘qSrr=)rPÚinclude©rprQrrrŠészhipify..csg|]}| ˆˆ¡‘qSrr=)rPrrWrrrŠês)r^r_rLr`rar)ÚPathr¸T)r,)rrƒz-Successfully preprocessed all matching files.r#) r1Úgetcwdr2r3r|r2ÚexitÚrstripÚshutilÚcopytreerÂror-rärhr>ÚpathlibrXÚrglobÚis_filer\r{rVÚnamer+rr%r'r*r3r…r )rQrRrLrSrprrr^rTr`r_rvrtrarUrurqZ all_files_setrHrXrZheader_include_dir_pathr2rsrZrrWrÚhipifyÊsn     ý    ÿ þ ý ü    ÿrb)rrrFF)F)F)FrNrOr‡rrPrFrTFFFN)\rKrJrWrÔr\r2r1r‡rZcuda_to_hip_mappingsrrÚtypingrrrr Úcollections.abcr r r{Z HipifyResultZHipifyFinalResultr&r Ú__annotations__ZPYTORCH_TEMPLATE_MAPrrr$r%r+ÚboolrVr\rorr…rÚcompilerÀrÇrÊrÐrÑZ RE_ASSERTrÓrÙrÛrÜrÝrêrkriròrjrõZ CAFFE2_TRIErZ PYTORCH_TRIEr ÚobjectrÚmappingrråÚitemsÚsrcrFÚdstZ meta_dataZ API_CAFFE2r4Z API_SPARSErZ API_PYTORCHrXr*rr)r+r,r-r.rzr:r<r>Z RE_INCLUDErÃrMrbrrrrÚs        úú , ÷  # (    H   =            ÷    . ñð