Sergei Lebedev 7dd887dc84 Updated the Triton lowering of jnp.transpose ...

Triton no longer restricts its transpose to just two axes.

PiperOrigin-RevId: 606188599

2024-02-12 02:42:33 -08:00

..

__init__.py

Generate Python bindings for the Triton MLIR dialect

2024-01-02 11:55:05 -08:00

BUILD

_triton_ext no longer links in MLIR C APIs

2024-02-07 03:39:29 -08:00

compat.py

Updated the Triton lowering of jnp.transpose

2024-02-12 02:42:33 -08:00

dialect.py

_triton_ext no longer links in MLIR C APIs

2024-02-07 03:39:29 -08:00

triton_dialect_capi.cc

Pallas now uses MLIR Python builders to lower to Triton IR

2024-01-11 13:33:26 -08:00

triton_dialect_capi.h

Add MLIR_CAPI_EXPORTED to Triton dialect C bindings

2024-02-07 02:14:50 -08:00

triton.td

Generate Python bindings for the Triton MLIR dialect

2024-01-02 11:55:05 -08:00