g%ddlmZmZmZmZmZmZmZddlm Z ddl m Z ddl m Z mZmZddlZddlZddlZddlZeZeZGdde ZejZGdd ZGd d Zd ZdS) )byrefc_charc_char_pc_intc_size_tc_void_pPOINTER)IntEnum)config) NvrtcErrorNvrtcCompilationErrorNvrtcSupportErrorNc>eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd S) 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_ERRORN/granary/edev/quantum/lib/python3.11/site-packages/numba/cuda/cudadrv/nvrtc.pyrrsWM !+,( !"#!",-)895673,.)!#r,rc4eZdZdZdZedZdZdS) 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"||_||_dSN)_nvrtc_handle)selfnvrtchandles r-__init__zNvrtcProgram.__init__+s  r,c|jSr1)r3r4s r-r6zNvrtcProgram.handle/s |r,cL|jr|j|dSdSr1)r3r2destroy_programr9s r-__del__zNvrtcProgram.__del__3s2 < . K ' ' - - - - - . .r,N)rrr__doc__r7propertyr6r<r+r,r-r/r/%sW X.....r,r/c @eZdZdZeeeeefeeeeeeeeefeeefeeeeefeeee feeefeeee feeefeeee feeefd Z dZ dZ dZ dZdZdZd Zd ZdS) 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}t|x|_} |d}n)#t $r}d|_td|d}~wwxYw|j D]c\}}t||}|d|_ |dd|_ tj|||dd}t|||ddddn #1swxYwY|jS)Nr) open_cudalibr5zNVRTC cannot be loadedr)funcnamec||}|tjkrt|tjkrC t|j}n#t $r d|d}YnwxYwd|d|}t |dS)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$GGG"7"9"99"k&???H-8-?-?-D #-HHH/G>C/G/G/G H#ID"H"HJ"H"HC",S//1@?sA AA) _nvrtc_lock_NVRTC__INSTANCEnumba.cuda.cudadrv.libsrLobject__new__OSErrorr _PROTOTYPESitemsgetattrrestypeargtypes functoolswrapssetattr) clsrLinstliberNprotorMrVs r-r[z NVRTC.__new__gs  6 6~%@@@@@@(.s(;(;;M&,w//CCMMM%)CN+,DEE1LM $(#3#9#9#;#;66KD%"3--D#(8DL$)!""IDM_T**15D 2 2 2 2+* 2D$ 5555= 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6@~s5/C6 AC6 A*A%%A**BC66C:=C:ct}t}|t|t||j|jfS)zB Get the NVRTC version as a tuple (major, minor). )rrArvalue)r4majorminors r- get_versionzNVRTC.get_versionsG %,,e 555{EK''r,c0t|tr|}t|tr|}t}|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!T4HHHD&)))r,cd|D}d|D}tt|z}||} ||jt||dS#t$rYdSwxYw)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. c6g|]}|Sr+)rr.0opts r- z)NVRTC.compile_program..s ;;;C3::<<;;;r,c,g|]}t|Sr+)rrxs r-r{z)NVRTC.compile_program..sDDDS8C==DDDr,FT)rlenrDr6r )r4programoptionsencoded_optionsoption_pointersc_options_type c_optionss r-compile_programzNVRTC.compile_programs<;7;;;DDODDD"S\\1"NO4    $ $W^S\\9 M M M5$   44 s)A!! A/.A/cT|t|jdS)z+ Destroy an NVRTC program. N)rCrr6)r4r~s r-r;zNVRTC.destroy_programs(   w~!6!677777r,ct}||jt|t |jz}||j||jS)z9 Get the compile log as a Python string. )rrIr6rrrkrJdecode)r4r~log_sizelogs r-get_compile_logzNVRTC.get_compile_logsi:: ##GNE(OODDD&)) 444y!!!r,ct}||jt|t |jz}||j||jS)z: Get the compiled PTX as a Python string. )rrEr6rrrkrFr)r4r~ptx_sizeptxs r-get_ptxz NVRTC.get_ptxsi:: W^U8__===&)) ---y!!!r,)rrrr= nvrtc_resultr rrsrrr]rXr[rnrurr;rrr+r,r-r@r@8sy&wwu~~wwu~~F ,]Hh$ggh&7&79J9JL!-ggm.D.DE!-mU ' 1 1 3)-9J9JK$mX>+M778;L;LM& x@$0#*78#4#4#6 ,]HEA!!KHJ!!!F(((***"$888 " " " " " " " "r,r@cNt}|||}|\}}d||}dtj}tjtjt} tj| } d| } ||| ddg} | || } | |}| rd|d|}t||rd|d|}tj |||}||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)rtrNccr5r~rlrmarchinclude cudadrv_pathnumba_cuda_path numba_includer compile_errorrrUrs r-compilers@ GGE""3--GLE5 7 7 7 7D-6+--G7??27??8#<#<==Lgool33O***MWmVVrsxNNNNNNNNNNNNNNNNNN9999999999    $ $ $ $ $' $ $ $in ........&Z"Z"Z"Z"Z"Z"Z"Z"z/////r,