Fix NcclSpace broadcast.

cedricchevalier19 · cedricchevalier19 · commit bac280b3d80b · 2025-12-08T21:21:20.000+01:00
Signed-off-by: Cedric Chevalier &lt;cedric.chevalier019@proton.me&gt;
diff --git a/src/KokkosComm/nccl/broadcast.hpp b/src/KokkosComm/nccl/broadcast.hpp
@@ -27,7 +27,7 @@ auto broadcast(const Kokkos::Cuda& space, View& v, int root, ncclComm_t comm) ->
 
   Req<NcclSpace> req{space.cuda_stream()};
   if (KC::is_contiguous(v)) {
-    ncclBcast(KC::data_handle(v), KC::span(v), datatype<NccSpace, T>, root, comm, space.cuda_stream());
+    ncclBcast(KC::data_handle(v), KC::span(v), datatype<NcclSpace, T>(), root, comm, space.cuda_stream());
   } else {
     Kokkos::abort("KokkosComm::Experimental::nccl::broadcast: unimplemented for non-contiguous views");
   }

Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ auto broadcast(const Kokkos::Cuda& space, View& v, int root, ncclComm_t comm) ->`
`27`	`27`
`28`	`28`	`Req<NcclSpace> req{space.cuda_stream()};`
`29`	`29`	`if (KC::is_contiguous(v)) {`
`30`		`- ncclBcast(KC::data_handle(v), KC::span(v), datatype<NccSpace, T>, root, comm, space.cuda_stream());`
	`30`	`+ ncclBcast(KC::data_handle(v), KC::span(v), datatype<NcclSpace, T>(), root, comm, space.cuda_stream());`
`31`	`31`	`} else {`
`32`	`32`	`Kokkos::abort("KokkosComm::Experimental::nccl::broadcast: unimplemented for non-contiguous views");`
`33`	`33`	`}`