Sergei Lebedev b904599b98 pl.debug_print no longer restricts values to be scalars ...

This allows printing arrays on Triton and soon on Mosaic GPU.

PiperOrigin-RevId: 675935666

2024-09-18 04:24:09 -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

pl.debug_print no longer restricts values to be scalars

2024-09-18 04:24:09 -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