U 9%e%@sddlmZmZmZmZmZmZmZddlm Z ddl m Z ddl m Z mZmZddlZddlZddlZddlZeZeZGddde ZeZGdd d ZGd d d Zd d ZdS))byrefc_charc_char_pc_intc_size_tc_void_pPOINTER)IntEnum)config) NvrtcErrorNvrtcCompilationErrorNvrtcSupportErrorNc@s<eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd S) NvrtcResultr N)__name__ __module__ __qualname__ NVRTC_SUCCESSZNVRTC_ERROR_OUT_OF_MEMORYZ$NVRTC_ERROR_PROGRAM_CREATION_FAILUREZNVRTC_ERROR_INVALID_INPUTZNVRTC_ERROR_INVALID_PROGRAMZNVRTC_ERROR_INVALID_OPTIONNVRTC_ERROR_COMPILATIONZ%NVRTC_ERROR_BUILTIN_OPERATION_FAILUREZ1NVRTC_ERROR_NO_NAME_EXPRESSIONS_AFTER_COMPILATIONZ/NVRTC_ERROR_NO_LOWERED_NAMES_BEFORE_COMPILATIONZ%NVRTC_ERROR_NAME_EXPRESSION_NOT_VALIDZNVRTC_ERROR_INTERNAL_ERRORrrW/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/numba/cuda/cudadrv/nvrtc.pyrsrc@s,eZdZdZddZeddZddZdS) 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. cCs||_||_dSN)_nvrtc_handle)selfnvrtchandlerrr __init__+szNvrtcProgram.__init__cCs|jSr")r$r%rrr r'/szNvrtcProgram.handlecCs|jr|j|dSr")r$r#destroy_programr)rrr __del__3szNvrtcProgram.__del__N)rrr__doc__r(propertyr'r+rrrr r!%s  r!c @seZdZdZeeeeefeeeeeeeeefeeefeeeeefeeee feeefeeee feeefeeee feeefd Z dZ ddZ ddZ dd Zd d Zd d ZddZddZdS)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 nvrtcGetPTXZnvrtcGetCUBINSizeZ nvrtcGetCUBINnvrtcGetProgramLogSizenvrtcGetProgramLogNc Cst|jdkrddlm}t||_}z |d}Wn2tk rj}zd|_td|W5d}~XYnX|j D]P\}}t ||}|d|_ |dd|_ t |||ddd}t|||qvW5QRX|jS) Nr) open_cudalibr&zNVRTC cannot be loadedr)funcnamecWsp||}|tjkrtnR|tjkrlzt|j}Wn tk rRd|d}YnXd|d|}t|dS)Nz"Unknown nvrtc_result (error code: )zFailed to call z: )rrr rr9 ValueErrorr )r8r9argserrorZ error_namemsgrrr checked_callxs  z#NVRTC.__new__..checked_call) _nvrtc_lock_NVRTC__INSTANCEZnumba.cuda.cudadrv.libsr7object__new__OSErrorr _PROTOTYPESitemsgetattrrestypeZargtypes functoolswrapssetattr) clsr7instliber9protor8r?rrr rCgs"      z NVRTC.__new__cCs,t}t}|t|t||j|jfS)zB Get the NVRTC version as a tuple (major, minor). )rr/rvalue)r%majorminorrrr get_versionszNVRTC.get_versioncCsLt|tr|}t|tr$|}t}|t|||dddt||S)z@ Create an NVRTC program with managed lifetime. rN) isinstancestrencode nvrtc_programr0rr!)r%srcr9r'rrr create_programs  zNVRTC.create_programcCsfdd|D}dd|D}tt|}||}z||jt||WdStk r`YdSXdS)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. cSsg|] }|qSr)rW.0optrrr sz)NVRTC.compile_program..cSsg|] }t|qSr)rr[rrr r^sFTN)rlenr2r'r )r%programoptionsZencoded_optionsZoption_pointersZc_options_typeZ c_optionsrrr compile_programs zNVRTC.compile_programcCs|t|jdS)z+ Destroy an NVRTC program. N)r1rr')r%r`rrr r*szNVRTC.destroy_programcCs<t}||jt|t|j}||j||jS)z9 Get the compile log as a Python string. )rr5r'rrrQr6decode)r%r`Zlog_sizelogrrr get_compile_logs  zNVRTC.get_compile_logcCs<t}||jt|t|j}||j||jS)z: Get the compiled PTX as a Python string. )rr3r'rrrQr4rc)r%r`Zptx_sizeptxrrr get_ptxs  z NVRTC.get_ptx)rrrr, nvrtc_resultrrrXrrrErArCrTrZrbr*rergrrrr r.8s8    $#  r.cCst}|||}|\}}d||}dtj}tjtjt} tj| } d| } ||| ddg} | || } | |}| rd|d|}t ||rd|d|}t |||}||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.rZr ZCUDA_INCLUDE_PATHospathdirnameabspath__file__rbrer warningswarnrg)rYr9ccr&r`rRrSarchincludeZ cudadrv_pathZnumba_cuda_pathZ numba_includeraZ compile_errorrdr>rfrrr compiles&         rt)ctypesrrrrrrrenumr Z numba.corer Znumba.cuda.cudadrv.errorr r r rIrj threadingrorXrhrLockr@r!r.rtrrrr s$