From b6493af6b7531f3d460d581e27415dd67b0bd6d3 Mon Sep 17 00:00:00 2001 From: chenjianping Date: Wed, 15 Apr 2020 14:18:50 +0000 Subject: [PATCH] fix gpu init fail --- cmake/options.cmake | 1 + mindspore/ccsrc/pipeline/init.cc | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/cmake/options.cmake b/cmake/options.cmake index 2f44e9a17c..3677418a98 100644 --- a/cmake/options.cmake +++ b/cmake/options.cmake @@ -64,6 +64,7 @@ endif() if (ENABLE_GPU) set(ENABLE_GPUQUE ON) + add_compile_definitions(ENABLE_GPU_COLLECTIVE) endif() if (ENABLE_GE) diff --git a/mindspore/ccsrc/pipeline/init.cc b/mindspore/ccsrc/pipeline/init.cc index 7c77949fda..b709199c87 100644 --- a/mindspore/ccsrc/pipeline/init.cc +++ b/mindspore/ccsrc/pipeline/init.cc @@ -29,7 +29,7 @@ #include "parallel/context.h" #include "parallel/device_manager.h" #include "parallel/costmodel_context.h" -#ifdef ENABLE_GPUQUE +#ifdef ENABLE_GPU_COLLECTIVE #include "device/gpu/distribution/collective_init.h" #else #include "device/gpu/distribution/collective_fake_init.h" @@ -300,7 +300,7 @@ PYBIND11_MODULE(_c_expression, m) { (void)py::class_>(m, "Oplib") .def(py::init()) .def("reg_op", &OpLib::RegOp, "Register op info."); -#ifdef ENABLE_GPUQUE +#ifdef ENABLE_GPU_COLLECTIVE (void)m.def("init_gpu_collective", &mindspore::device::gpu::CollectiveInitializer::InitCollective, "Init gpu collective communication mode."); (void)m.def("finalize_gpu_collective", &mindspore::device::gpu::CollectiveInitializer::FinalizeCollective,