Sergei Lebedev e90336947a Pulled scratch_shapes into GridSpec ...

It is supported by Mosaic TPU and Mosaic GPU and unsupported by Triton.

PiperOrigin-RevId: 675950199

2024-09-18 05:26:21 -07:00

..

tpu

Merge pull request #23484 from justinjfu:pallas_prefetch_docs

2024-09-09 07:33:57 -07:00

async_note.md

[Pallas] Add design note for async ops on TPU

2024-09-17 12:45:29 -07:00

CHANGELOG.md

Pulled scratch_shapes into GridSpec

2024-09-18 05:26:21 -07:00

design.md

Changed `pl.BlockSpec to accept block_shape before index_map`

2024-07-01 14:26:08 -07:00

grid_blockspec.md

[pallas] Enable check for GPU lowering that tensor sizes are power of 2

2024-08-05 02:34:21 -07:00

index.rst

[Pallas] Add design note for async ops on TPU

2024-09-17 12:45:29 -07:00

quickstart.ipynb

Changed `pl.BlockSpec to accept block_shape before index_map`

2024-07-01 14:26:08 -07:00

quickstart.md

pre-commit: update hooks & pin using hashes

2024-08-27 15:23:13 -07:00