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 06d3280890 Inlined some Triton-specific abstractions 
* tensor is now just a container for dtype/shape with no extra methods;
* constexpr is not used;
* all APIs assume that the arguments are tensors and do no ->tensor conversion.

PiperOrigin-RevId: 603894852
2024-02-03 00:56:47 -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 some Triton-specific abstractions
2024-02-03 00:56:47 -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