Sharad Vikram 8fbcfce2dd [Pallas] Enable interpreter mode as default lowering for CPU ...

PiperOrigin-RevId: 580700740

2023-11-08 16:35:31 -08:00

..

mosaic

Small Pallas bug fixes.

2023-11-07 17:55:19 -08:00

triton

[Pallas] Preserve order of grid provided to pallas_call, append and reverse vmapped dimensions

2023-11-07 18:23:07 -08:00

__init__.py

…

BUILD

Load Python rules from rules_python

2023-08-24 10:22:57 -07:00

core.py

[Pallas TPU] Add support for hoisted scratch spaces

2023-10-24 17:29:54 -07:00

indexing.py

[pallas] Simplify Slice.from_slice code and add check for Slice.size.

2023-10-13 01:36:28 -07:00

pallas_call.py

[Pallas] Enable interpreter mode as default lowering for CPU

2023-11-08 16:35:31 -08:00

primitives.py

[pallas] dot fixes.

2023-10-16 12:35:43 -07:00

utils.py

[Pallas] Add support for lax.fori_loop to Mosaic lowering

2023-08-04 13:43:43 -07:00