jax
1bf70fbb - [pallas:mosaic_gpu] `copy_gmem_to_smem` no longer requires `barrier` to be a keyword argument

Commit
1 year ago
[pallas:mosaic_gpu] `copy_gmem_to_smem` no longer requires `barrier` to be a keyword argument ... because there really isn't any reason to require that. PiperOrigin-RevId: 698116984
Author
Parents
Loading