:j%ddlmZmZmZmZmZmZmZddlm Z ddl m Z ddl m Z mZmZddlZddlZddlZddlZeZeZGdde Zej.ZGdd ZGd d Zd Zy) )byrefc_charc_char_pc_intc_size_tc_void_pPOINTER)IntEnum)config) NvrtcErrorNvrtcCompilationErrorNvrtcSupportErrorNc<eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zy ) NvrtcResultr N)__name__ __module__ __qualname__ NVRTC_SUCCESSNVRTC_ERROR_OUT_OF_MEMORY$NVRTC_ERROR_PROGRAM_CREATION_FAILURENVRTC_ERROR_INVALID_INPUTNVRTC_ERROR_INVALID_PROGRAMNVRTC_ERROR_INVALID_OPTIONNVRTC_ERROR_COMPILATION%NVRTC_ERROR_BUILTIN_OPERATION_FAILURE1NVRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATION/NVRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATION%NVRTC_ERROR_NAME_EXPRESSION_NOT_VALIDNVRTC_ERROR_INTERNAL_ERRORm/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/numba/cuda/cudadrv/nvrtc.pyrrsIM !+,( !"#!",-)895673,.)!#r,rc,eZdZdZdZedZdZy) NvrtcProgramz A class for managing the lifetime of nvrtcProgram instances. Instances of the class own an nvrtcProgram; when an instance is deleted, the underlying nvrtcProgram is destroyed using the appropriate NVRTC API. c ||_||_yN)_nvrtc_handle)selfnvrtchandles r-__init__zNvrtcProgram.__init__+s  r,c|jSr1)r3r4s r-r6zNvrtcProgram.handle/s ||r,cT|jr|jj|yyr1)r3r2destroy_programr9s r-__del__zNvrtcProgram.__del__3s << KK ' ' - r,N)rrr__doc__r7propertyr6r<r+r,r-r/r/%s% .r,r/c eZdZdZeeeeefeeeeeeeeefeeefeeeeefeeee feeefeeee feeefeeee feeefd Z dZ dZ dZ dZdZdZd Zd Zy) NVRTCaB Provides a Pythonic interface to the NVRTC APIs, abstracting away the C API calls. The sole instance of this class is a process-wide singleton, similar to the NVVM interface. Initialization is protected by a lock and uses the standard (for Numba) open_cudalib function to load the NVRTC library. ) nvrtcVersionnvrtcCreateProgramnvrtcDestroyProgramnvrtcCompileProgramnvrtcGetPTXSize nvrtcGetPTXnvrtcGetCUBINSize nvrtcGetCUBINnvrtcGetProgramLogSizenvrtcGetProgramLogNct5|jddlm}tj |x|_} |d}|jjD]Q\}}t||}|d|_ |dd|_ tj|||dd}t|||Sddd|jS#t $r}d|_td|d}~wwxYw#1swY|jSxYw)Nr) open_cudalibr5zNVRTC cannot be loadedr)funcnamec||}|tjk(r t|tjk7r) t|j}d|d|}t |y#t $r d|d}Y%wxYw)Nz"Unknown nvrtc_result (error code: )zFailed to call z: )rr%r rrN ValueErrorr )rMrNargserror error_namemsgs r- checked_callz#NVRTC.__new__..checked_callxs $d  K$G$GG"7"99"k&?&??H-8-?-D-D %4D6J<"HC",S/1@$.H/>>CWA/G HsA A21A2) _nvrtc_lock_NVRTC__INSTANCEnumba.cuda.cudadrv.libsrLobject__new__OSErrorr _PROTOTYPESitemsgetattrrestypeargtypes functoolswrapssetattr) clsrLinstliberNprotorMrVs r-r[z NVRTC.__new__gs  6~~%@(.s(;;M&w/C $(#3#3#9#9#;6KD%"3-D#(8DL$)!"IDM__T*15D 2+ 2D$ 5'6 6@~~5M%)CN+,DE1LM  6@~~s./C%CA.C% C" CC""C%%C9ct}t}|jt|t||j|jfS)zB Get the NVRTC version as a tuple (major, minor). )rrArvalue)r4majorminors r- get_versionzNVRTC.get_versions= %,e 5{{EKK''r,ct|tr|j}t|tr|j}t}|j t |||dddt ||S)z@ Create an NVRTC program with managed lifetime. rN) isinstancestrencode nvrtc_programrBrr/)r4srcrNr6s r-create_programzNVRTC.create_programs` c3 **,C dC ;;=D f sD!T4HD&))r,c|Dcgc]}|j}}|Dcgc] }t|}}tt|z}||} |j|jt||ycc}wcc}w#t $rYywxYw)z Compile an NVRTC program. Compilation may fail due to a user error in the source; this function returns ``True`` if there is a compilation error and ``False`` on success. FT)rrrlenrDr6r )r4programoptionsoptencoded_optionsoption_pointersc_options_type c_optionss r-compile_programzNVRTC.compile_programs4;;C3::<;;4CDS8C=DD"S\1"O4    $ $W^^S\9 M <D %  sA5A:&A?? B  B cL|jt|jy)z+ Destroy an NVRTC program. N)rCrr6)r4rxs r-r;zNVRTC.destroy_programs   w~~!67r,ct}|j|jt|t |j z}|j |j||j jS)z9 Get the compile log as a Python string. )rrIr6rrrkrJdecode)r4rxlog_sizelogs r-get_compile_logzNVRTC.get_compile_logs[: ##GNNE(OD&) 4yy!!r,ct}|j|jt|t |j z}|j |j||j jS)z: Get the compiled PTX as a Python string. )rrEr6rrrkrFr)r4rxptx_sizeptxs r-get_ptxz NVRTC.get_ptxs[: W^^U8_=&) -yy!!r,)rrrr= nvrtc_resultr rrsrrr]rXr[rnrurr;rrr+r,r-r@r@8s&wu~wu~F ,]Hh$gh&79JL!-gm.DE!-mU ' 1 3)-9JK$mX>+M78;LM& x@$0#*8#4#6 ,]HEA!KHJ!F(*"$8 " "r,r@c&t}|j||}|\}}d||}dtj}tj j tj jt} tj j | } d| } ||| ddg} |j|| } |j|}| rd|d|}t||rd|d|}tj||j|}||fS)a~ Compile a CUDA C/C++ source to PTX for a given compute capability. :param src: The source code to compile :type src: str :param name: The filename of the source (for information only) :type name: str :param cc: A tuple ``(major, minor)`` of the compute capability :type cc: tuple :return: The compiled PTX and compilation log :rtype: tuple z--gpu-architecture=compute_z-Iz-rdctruez+NVRTC Compilation failure whilst compiling z: z$NVRTC log messages whilst compiling )r@rur CUDA_INCLUDE_PATHospathdirnameabspath__file__rrr warningswarnr)rtrNccr5rxrlrmarchinclude cudadrv_pathnumba_cuda_path numba_includery compile_errorrrUrs r-compilers GE""3-GLE5 (w 7D6++,-G77??277??8#<=Lggool3O)*MWmVVrsrNNN99    $' $inn ..&Z"Z"z/r,