jax
a61d8002 - [Mosaic GPU] Relax TMEM stride constraints on dimensions of size 1

Commit

1 year ago

[Mosaic GPU] Relax TMEM stride constraints on dimensions of size 1 Strides along those dimensions don't affect anything. PiperOrigin-RevId: 723968994

References

Author

apaszke

Committer

Google-ML-Automation

Parents