diff --git a/oneflow/core/job/runtime.cpp b/oneflow/core/job/runtime.cpp index 1eaeb3d67fa..068484c7a78 100644 --- a/oneflow/core/job/runtime.cpp +++ b/oneflow/core/job/runtime.cpp @@ -70,10 +70,9 @@ Runtime::Runtime( Singleton::Get()->AddPlan(plan); collective_boxing_scheduler_plan_token_ = Singleton::Get()->AddPlan(plan); - // #ifdef WITH_CUDA - // Singleton::Get()->CreateCommFromPlan(plan); - // #endif // WITH_CUDA +#ifdef WITH_CUDA Singleton::Get()->CreateCommFromPlan(plan); +#endif // WITH_CUDA } std::vector source_tasks; source_tasks.reserve(plan.task().size());