jax

ae0ea1df - [pallas:sc] Allowed scalar loads and stores to/from VMEM

[pallas:sc] Allowed scalar loads and stores to/from VMEM Apparently, the SC compiler does not support copies from HBM to SMEM on the vector subcore, so the only way of passing scalars is via VMEM. PiperOrigin-RevId: 817599372