Sergei Lebedev f15cad4651 Lower a subset of math primitives directly to Triton IR ...

Note that all primitives are now lowered to libdevice calls. Previously,
some of them were lowered to the MLIR arith dialect, and some to libdevice
calls, without any apparent reason for doing so.

PiperOrigin-RevId: 601259707

2024-01-24 15:55:09 -08:00

..

__init__.py

Generate Python bindings for the Triton MLIR dialect

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

_triton_ext.cc

Pallas now uses MLIR Python builders to lower to Triton IR

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

_triton_ext.pyi

Pallas now uses MLIR Python builders to lower to Triton IR

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

BUILD

Pallas now uses MLIR Python builders to lower to Triton IR

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

compat.py

Lower a subset of math primitives directly to Triton IR

2024-01-24 15:55:09 -08:00

dialect.py

Pallas now uses MLIR Python builders to lower to Triton IR

2024-01-11 13:33:26 -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

Pallas now uses MLIR Python builders to lower to Triton IR

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

triton.td

Generate Python bindings for the Triton MLIR dialect

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