jax
231b7351 - [Pallas:MGPU] Raise when loading/storing from/to a non-transposed ref using a transposed layout.

Commit
77 days ago
[Pallas:MGPU] Raise when loading/storing from/to a non-transposed ref using a transposed layout. PiperOrigin-RevId: 914727860
Author
Parents
Loading