sh%SSKJrJrJrJrJrJrJr SSKJ r SSK J r SSK J r JrJr SSKrSSKrSSKrSSKr\r\r"SS\ 5r\R."5r"SS 5r"S S 5rS rg) )byrefc_charc_char_pc_intc_size_tc_void_pPOINTER)IntEnum)config) NvrtcErrorNvrtcCompilationErrorNvrtcSupportErrorNcD\rSrSrSrSrSrSrSrSr Sr S r S r S r S rS rSrg) NvrtcResultr N)__name__ __module__ __qualname____firstlineno__ 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_ERROR__static_attributes__rl/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/cuda/cudadrv/nvrtc.pyrrsIM !+,( !"#!",-)895673,.)!#r/rc4\rSrSrSrSr\S5rSrSr g) NvrtcProgram%z 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. cXlX lgN)_nvrtc_handle)selfnvrtchandles r0__init__NvrtcProgram.__init__+s   r/cUR$r5)r7r8s r0r:NvrtcProgram.handle/s ||r/c^UR(aURRU5 ggr5)r7r6destroy_programr>s r0__del__NvrtcProgram.__del__3s << KK ' ' - r/)r7r6N) rrr r!__doc__r;propertyr:rBr.rr/r0r2r2%s% .r/r2c \rSrSrSr\\"\5\"\54\\\ \ \\"\ 5\"\ 54\\"\54\\\\"\ 54\\\"\ 54\\\ 4\\\"\ 54\\\ 4\\\"\ 54\\\ 4S. r Sr Sr SrSrSrS rS rS rS rg) NVRTC8a* 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 nvrtcGetCUBINnvrtcGetProgramLogSizenvrtcGetProgramLogNc[ URcSSKJn [R U5=UlnU"S5nURR5HPupV[X55nUSUl USSUl [R"U5XuS.Sj5n[X%U5 MR SSS5 UR$![ anSUl[S5UeSnAff=f!,(df  UR$=f)Nr) open_cudalibr9zNVRTC cannot be loadedr)funcnamecU"U6nU[R:Xa [5eU[R:wa)[U5RnSUSU3n[ U5eg![ a SUS3nN&f=f)Nz"Unknown nvrtc_result (error code: )zFailed to call z: )rr(r r"rV ValueErrorr )rUrVargserror error_namemsgs r0 checked_call#NVRTC.__new__..checked_callxs $d  K$G$GG"7"99"k&?&??H-8-?-D-D %4D6J<"HC",S/1@$.H/>>CWA/G HsA""A54A5) _nvrtc_lock_NVRTC__INSTANCEnumba.cuda.cudadrv.libsrTobject__new__OSErrorr _PROTOTYPESitemsgetattrrestypeargtypes functoolswrapssetattr) clsrTinstliberVprotorUr^s r0rd NVRTC.__new__gs ~~%@(.s(;;M&w/C $(#3#3#9#9#;KD"3-D#(8DL$)!"IDM__T*15 2+ 2D 5'$<@~~5M%)CN+,DE1LM [@~~s/0C'CA.C' C$ CC$$C'' Dc[5n[5nUR[U5[U55 URUR4$)z2 Get the NVRTC version as a tuple (major, minor). )rrIrvalue)r8majorminors r0 get_versionNVRTC.get_versions= %,e 5{{EKK''r/c[U[5(aUR5n[U[5(aUR5n[5nUR [ U5XSSS5 [ X5$)z0 Create an NVRTC program with managed lifetime. rN) isinstancestrencode nvrtc_programrJrr2)r8srcrVr:s r0create_programNVRTC.create_programsb c3  **,C dC ;;=D f s!T4HD))r/c(UVs/sHo3R5PM nnUVs/sHn[U5PM nn[[U5-nU"U6nURUR[U5U5 gs snfs snf![ a gf=f)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)r}rlenrLr:r )r8programoptionsoptencoded_optionsoption_pointersc_options_type c_optionss r0compile_programNVRTC.compile_programs4;;7C::<7;4CDOS8C=OD"S\1"O4    $ $W^^S\9 M <D %  sA:A?&B BBcLUR[UR55 g)z Destroy an NVRTC program. N)rKrr:)r8rs r0rANVRTC.destroy_programs   w~~!67r/c[5nURUR[U55 [UR -"5nUR URU5 UR R5$)z) Get the compile log as a Python string. )rrQr:rrrurRdecode)r8rlog_sizelogs r0get_compile_logNVRTC.get_compile_logs[: ##GNNE(OD&) 4yy!!r/c[5nURUR[U55 [UR -"5nUR URU5 UR R5$)z* Get the compiled PTX as a Python string. )rrMr:rrrurNr)r8rptx_sizeptxs r0get_ptx NVRTC.get_ptxs[: W^^U8_=&) -yy!!r/r)rrr r!rD nvrtc_resultr rr~rrrfrardrxrrrArrr.rr/r0rGrG8s&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/rGc4[5nURX5nUupVSUU3nS[R3n[R R [R R[55n [R R U 5n SU 3n XxU SS/n URXL5n URU5nU (aSUSU3n[U5eU(aSUSU3n[R"U5 URU5nUU4$)aV 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 )rGrr CUDA_INCLUDE_PATHospathdirnameabspath__file__rrr warningswarnr)rrVccr9rrvrwarchinclude cudadrv_pathnumba_cuda_path numba_includer compile_errorrr]rs r0compilers GE""3-GLE (w 7D6++,-G77??277??8#<=Lggool3O)*MmVVrspNNN99    $' $nn ..&Z"Z"z/r/