
    uki                         d Z ddl ddl ddl 	 ddlmZ ej                  j                  e
de
j                  d              y# e$ r	 ddlmZ Y ;w xY w)zPython bindings for the MLIR Mosaic GPU dialect.

Note: this file *must* be called `mosaic_gpu.py`, in order to match the dialect
name. Otherwise, MLIR is unable to find the module during dialect search.
    )*)_cextN.)__doc__-jaxlib.mosaic.dialect.gpu._mosaic_gpu_gen_ops/jaxlib.mosaic.dialect.gpu._mosaic_gpu_gen_enums&jaxlib.mlir._mlir_libs._mosaic_gpu_ext jaxlib.mlir.dialects._ods_commonr   ImportErrormlir.dialects._ods_commonglobalsappend_dialect_search_prefix__name__rfind     Z/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jaxlib/mosaic/python/mosaic_gpu.py<module>r      sV    < = 4.4  * *84HX^^C5H+I J  .-.s   A AA