
    uki                        d dl mZ ddlmZ  edd      Z edd      Zer ej                  dej                         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ej                  Zer ej                  dej                         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ej                  Zyy)    )
xla_client   )import_from_plugincuda_tritonrocmtriton_kernel_callCUDA)platformROCMN)jaxlibr   plugin_supportr   _cuda_triton_hip_tritonregister_custom_call_targetget_custom_callTritonKernelCallTritonAutotunedKernelCallTritonKernelcreate_array_parametercreate_scalar_parameterget_compute_capabilityget_arch_detailsget_serialized_metadata     L/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jaxlib/gpu_triton.py<module>r      sJ    .!&)4 3(*((L88: "22*DD**,'>>(@@'>>!22 00/(@@(*((K779 !11)CC)),&=='??&== 11///'?? r   