jax
6eb71e90 - [Pallas/interpreter] Add a prototype for a GPU kernel interpreter.

Commit
29 days ago
[Pallas/interpreter] Add a prototype for a GPU kernel interpreter. Current limitiations: - Only trivial grids are supported. - An arbitrary number of threads, along a single axis, is supported. - Primitives for memory transfers, synchronization or `wgmma` are not supported yet. PiperOrigin-RevId: 855618516
Parents
Loading