Christos Perivolaropoulos eeccc67c0b [mgpu] Debug print arrays. ...

PiperOrigin-RevId: 734576543

2025-03-07 08:58:25 -08:00

..

fuser

[Pallas/Fuser] Add support for custom_call_jvp/pjit to push_block_spec

2025-03-03 17:43:13 -08:00

mosaic

[Pallas] Adds a simple dynamic race detector for TPU interpret mode.

2025-03-05 15:15:21 -08:00

mosaic_gpu

[mgpu] Debug print arrays.

2025-03-07 08:58:25 -08:00

triton

[pallas:triton] Emit a better error message for matmul with non-2D operands

2025-03-04 05:46:29 -08:00

__init__.py

[Pallas] Upstream pallas to JAX

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

BUILD

[JAX] Allow pallas to accept scalar shape semaphores.

2025-02-14 23:20:47 -08:00

core.py

[Pallas] Add name parameter to core_map

2025-02-26 18:59:01 -08:00

cost_estimate.py

[better_errors] Continue adding debug info to Jaxprs (step 7)

2025-02-09 18:14:33 +02:00

helpers.py

[Pallas TPU] Add helpers for writing collectives

2025-02-04 15:39:10 -08:00

hlo_interpreter.py

[better_errors] Port the Pallas debug info mechanisms to the new JAX DebugInfo.

2025-02-25 14:43:17 +01:00

pallas_call.py

[Pallas] Add option for async DMAs in the new TPU interpret mode

2025-02-25 18:19:20 -08:00

primitives.py

[pallas] Add support for pl.dot with int8 inputs.

2025-03-06 04:08:04 -08:00

utils.py

[Pallas TPU] Add helpers for writing collectives

2025-02-04 15:39:10 -08:00