·
6 commits
to dev-2.9.6-for-jax
since this release
As new operations ncclSend, SendRecv, etc. have been introduced into NCCL library since 2.7.* and topo algorithm of devices sorting have been optimized from that time, I spent efforts to migrate Nvidia NCCL 2.9.6-1 to macOS 10.13.6 now.
Currently you still need a workaround mentioned in https://github.com/llv22/nccl-osx/blob/dev-2.9.6-for-jax/MIGRATION.md to query device status. But for the next release 2.9.7, native API query based on CUDA and system calls will replace current workaround.