mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/triton
History
Sergei Lebedev 0596c804cb Inlined all builder.create_* calls 
The lowering code now directly calls into the auto-generated MLIR bindings
for the respective dialects.

PiperOrigin-RevId: 604260837
2024-02-05 03:00:33 -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
Inlined all builder.create_* calls
2024-02-05 03:00:33 -08:00
dialect.py
Inlined all builder.create_* calls
2024-02-05 03:00:33 -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