MXNet: Fix "mixed" build

Signed-off-by: Max H. Gerlach <git@maxgerlach.de>
horovod · Jul 22, 2022 · 16b472e · 16b472e
1 parent 54f93bc
commit 16b472e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/horovod/mxnet/mpi_ops.cc b/horovod/mxnet/mpi_ops.cc
@@ -764,7 +764,7 @@ extern "C" int horovod_mxnet_reducescatter_async(NDArray* const* inputs,
   MX_API_BEGIN();
 
 #if HAVE_CUDA && !HOROVOD_GPU_REDUCESCATTER
-  if (IsTensorOnCPU(input) && IsTensorOnCPU(output)) {
+  if (IsTensorOnCPU(inputs[0]) && IsTensorOnCPU(outputs[0])) {
     PushHorovodOperation(OperationType::REDUCESCATTER, inputs, outputs, name,
                          priority, num_tensors, process_set_id, -1, false);
   } else {