
    Spf	                     >   d dl Z d dlmZ dD ]&Z	  e j        e dd          Z n# e$ r dZY #w xY wer] ej        de                                d	           ej	        Z	ej
        Z
ej        Zej        Zej        Zej        Zej        Zej        Zd
D ]&Z	  e j        e dd          Z n# e$ r dZY #w xY wer_ ej        de                                d	           ej	        Z	ej
        Z
ej        Zej        Zej        Zej        Zej        Zej        ZdS dS )    N)
xla_client)z.cudajax_cuda12_pluginz._tritonjaxlib)packagetriton_kernel_callCUDA)platform)z.rocmjax_rocm60_pluginROCM)	importlibr   r   cuda_module_nameimport_module_cuda_tritonImportErrorregister_custom_call_targetget_custom_callTritonKernelCallTritonAutotunedKernelCallTritonKernelcreate_array_parametercreate_scalar_parameterget_compute_capabilityget_serialized_metadatarocm_module_name_hip_triton     Q/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jaxlib/gpu_triton.py<module>r      s            6 
 

*9*%%%x  L 
E 
   LLL
  A(*(L88::    "2*D*,'>(@'> 0/(@6 
 

)))%%%x  K 
E 
   KKK
  @(*(K7799    !1)C),&='?&=//'?@ @s   '11B//B98B9