U yʹjq @sUddlZddlZddlZddlZddlZddlZddlmZddlm Z GdddZ iZ ee e fe d<e edd d Ze d d d ZeddZdS)N)Dict) DriverBasec@seZdZeddddZddZddZd d Zed d d Z ddZ ddZ ddZ ddZ ddZddZddZeedddZdS) BaseBackendN) device_typereturncCs ||_dSNr)selfrr T/var/www/html/bwave/python/venv/lib/python3.8/site-packages/triton/common/backend.py__init__szBaseBackend.__init__cCstdS)zZ Custom the arch, extern_libs and stages per backend specific requirement NNotImplementedError)r archZ extern_libsZstagesr r r add_stagesszBaseBackend.add_stagescCstdS)zZ Custom the ir, module, metadata and asm per backend specific requirement Nr)r ZirZ cur_moduleZ next_modulemetadataasmr r r add_meta_infoszBaseBackend.add_meta_infocCstdS)z2 Return a callable to load binary Nrr r r r get_load_binary_fnszBaseBackend.get_load_binary_fn)rcCstdS)zW Get the backend driver. Please refer to "DriverBase" for more details Nrrr r r get_driver#szBaseBackend.get_drivercCstdS)z/ Get stream for current device Nrrr r r get_stream)szBaseBackend.get_streamcCstdSrrr Zdevicer r r get_device_properties/sz!BaseBackend.get_device_propertiescCstdS)z$ Get current device Nrrr r r get_current_device2szBaseBackend.get_current_devicecCstdS)z8 Set current device as the given device Nrrr r r set_current_device8szBaseBackend.set_current_devicecCstdSrrrr r r get_kernel_bin>szBaseBackend.get_kernel_bincCstdS)zA Generate the launcher stub to launch the kernel Nr)r name signature constantsr r r make_launcher_stubAszBaseBackend.make_launcher_stubcKstdS)z= Get the architecture descriptor the backend Nr)r kwargsr r r get_architecture_descriptorGsz'BaseBackend.get_architecture_descriptorr cCs||Srr )clsrr r r create_backendMszBaseBackend.create_backend)__name__ __module__ __qualname__strr rrrrrrrrrrr!r# classmethodr%r r r r r sr _backendsrZ backend_clscCs|tkr||t|<dSr)r+r%r,r r r register_backendUsr-r cCsj|tkrVd|}tjj|tjdrRztj|tjdWqVtk rNYdSXndS|tkrft|SdS)Nz...third_party.)package)r+ importlibutil find_spec__spec__r import_module Exception)rZdevice_backend_package_namer r r get_backendZs  r5cCstjtjttj}tjddtj|ddddg}|D]x}|dd}tj |r>tj |r>t j |d gt j d }|dk r>tjd |d tjd }|dk r>||dfSq>tddS)NZTRITON_PTXAS_PATHZ third_partycudabinptxas rz --version)stderrz.*release (\d+\.\d+).*zutf-8)flagszCannot find ptxas)ospathjoindirname__file__pardirenvirongetsplitexistsisfile subprocess check_outputSTDOUTresearchdecode MULTILINEgroup RuntimeError)base_dirpathsr9Z ptxas_binresultversionr r r path_to_ptxasgs rV) functoolsr/Zimportlib.utilr>rLrItypingrZruntime.driverrrr+r)__annotations__typer-r5 lru_cacherVr r r r s  E