fixing multidevice build

jjsjann123 · jjsjann123 · commit 5f8c84b55998 · 2024-02-12T14:16:49.000-08:00
API change coming from pytorch/pytorch#119421
diff --git a/csrc/multidevice/communication.cpp b/csrc/multidevice/communication.cpp
@@ -236,7 +236,8 @@ c10::intrusive_ptr<c10d::Work> Reduce::post(
 #ifdef USE_C10D_NCCL
   auto nccl_backend = dynamic_cast<c10d::ProcessGroupNCCL*>(team_backend.get());
   if (nccl_backend) {
-    return nccl_backend->_reduce_oop(buf, params_.src_bufs, options);
+    return nccl_backend->_reduce_oop(
+        buf.at(0), params_.src_bufs.at(0), options);
   }
 #endif
   if (comm.deviceId() == params_.root) {
