George HotzandGitHub b30c7e00d4 support 2d on UNSHARD (kimi) (#17285) ...

* support 2d on UNSHARD

* fixes

* Fix test and spec

* single barrier

* 2d sharding works for devices too

* cleanups

* no _rewrap

2026-07-29 12:01:59 -07:00

96 KiB

Raw Permalink History

View Raw