jax
54fd738e - Add SMEM as a supported Pallas output memory space.

Commit

1 year ago

Add SMEM as a supported Pallas output memory space. PiperOrigin-RevId: 712144883

References

#25689 - Add SMEM as a supported Pallas output memory space.

#193 - CI: 01/06/25 upstream sync

Author

Google-ML-Automation

Google-ML-Automation

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading