Benjamin Chetioui 63e59c5fd7 [Mosaic GPU] Ensure that the dialect module can be loaded successfully. ...

This requires that the file providing the bindings has the same name as the
dialect it defines, since dialect search looks for a module path of the form
`<prefix>.<dialect namespace>`.

PiperOrigin-RevId: 693241875

2024-11-05 00:47:21 -08:00

..

mlir

Partially rolling forward #22998

2024-09-02 04:44:47 -07:00

__init__.py

[Mosaic GPU] Ensure that the dialect module can be loaded successfully.

2024-11-05 00:47:21 -08:00

BUILD

[Mosaic GPU] Add Python bindings for the Mosaic GPU MLIR dialect.

2024-10-31 02:47:30 -07:00

mosaic_gpu.py

Make CuDNN SDPA API work with JAX with a CUDA plugin configuration.

2024-06-06 12:09:19 -04:00

triton.py

Do not import from lowering in tests/pallas/pallas_test.py

2024-05-10 14:25:17 -07:00