diff --git a/src/rccl_wrap.cc b/src/rccl_wrap.cc index 2b1bf1f0a8..2ab0cac8ea 100644 --- a/src/rccl_wrap.cc +++ b/src/rccl_wrap.cc @@ -378,7 +378,7 @@ ncclResult_t commSetUnrollFactor(struct ncclComm* comm) { else comm->unroll = NCCL_UNROLL_4; - INFO(NCCL_INIT, "RCCL Unroll Factor (pre-set): %d", comm->unroll+1); + INFO(NCCL_INIT, "RCCL Unroll Factor (pre-set): %d", (int) (pow(2.0, (double)comm->unroll))); return ncclSuccess; }