mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 12:26:07 +00:00
22 lines
341 B
ReStructuredText
22 lines
341 B
ReStructuredText
|
``jax.experimental.pallas.triton`` module
|
||
|
|
=========================================
|
||
|
|
|
||
|
|
.. automodule:: jax.experimental.pallas.triton
|
||
|
|
|
||
|
|
Classes
|
||
|
|
-------
|
||
|
|
|
||
|
|
.. autosummary::
|
||
|
|
:toctree: _autosummary
|
||
|
|
|
||
|
|
TritonCompilerParams
|
||
|
|
|
||
|
|
Functions
|
||
|
|
---------
|
||
|
|
|
||
|
|
.. autosummary::
|
||
|
|
:toctree: _autosummary
|
||
|
|
|
||
|
|
approx_tanh
|
||
|
|
debug_barrier
|
||
|
|
elementwise_inline_asm
|