Skip to content

[Mosaic GPU] Relax TMEM stride constraints on dimensions of size 1 #3013

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

[Mosaic GPU] Relax TMEM stride constraints on dimensions of size 1 #3013

TPU test (jaxlib=head, v5e-8)

succeeded Feb 6, 2025 in 7m 9s