Peter Hawkins 7f4ef63cd8 Run pyupgrade --py310-plus. ...

Also apply manual fixes to import sorting and unused imports.

2024-06-26 16:10:18 -04:00

..

mosaic

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00

mosaic_gpu

Removed unused which_linear= param from pallas_call_p

2024-06-18 11:31:54 -07:00

triton

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00

__init__.py

[Pallas] Upstream pallas to JAX

2023-08-01 16:43:13 -07:00

BUILD

Removed the double re-exporting of Pallas GPU/TPU APIs

2024-06-10 05:59:09 -07:00

core.py

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00

pallas_call.py

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00

primitives.py

[Pallas] Make num_programs return an int if the grid is not dynamic

2024-06-17 15:18:40 -07:00

utils.py

Add support to pipeline emitter for shapes that don't perfectly divide the block shapes

2024-06-18 14:26:31 -07:00