Browse Source

fix mac undefined symbol error

r1.7
xulei 4 years ago
parent
commit
74e3552547
2 changed files with 3 additions and 1 deletions
  1. +1
    -0
      mindspore/ccsrc/runtime/graph_scheduler/CMakeLists.txt
  2. +2
    -1
      mindspore/ccsrc/runtime/graph_scheduler/graph_compiler.h

+ 1
- 0
mindspore/ccsrc/runtime/graph_scheduler/CMakeLists.txt View File

@@ -12,4 +12,5 @@ endif()

set_property(SOURCE ${GRAPH_SCHEDULER_SRC_LIST}
PROPERTY COMPILE_DEFINITIONS SUBMODULE_ID=mindspore::SubModuleId::SM_RUNTIME_FRAMEWORK)

add_library(_mindspore_runtime_graph_scheduler_obj OBJECT ${GRAPH_SCHEDULER_SRC_LIST})

+ 2
- 1
mindspore/ccsrc/runtime/graph_scheduler/graph_compiler.h View File

@@ -29,6 +29,7 @@
#include "backend/common/session/session_basic.h"
#include "backend/common/session/session_factory.h"
#include "ir/tensor.h"
#include "include/backend/visible.h"

namespace mindspore {
using device::DeviceContext;
@@ -55,7 +56,7 @@ using KernelMapPosition = std::map<KernelWithIndex, std::vector<size_t>, session
// The origin parameters order is used to correspond to the input args.
// The origin outputs order is used to correspond to the output args.
// The need_erase means need erase this GraphCompilerInfo object after run actor set.
struct GraphCompilerInfo {
struct BACKEND_EXPORT GraphCompilerInfo {
GraphCompilerInfo(const std::vector<KernelGraphPtr> &graphs, const std::vector<DeviceContext *> &device_contexts,
const std::vector<std::vector<int64_t> *> &tensors_mask,
const std::vector<std::vector<TensorPtr> *> &input_tensors,


Loading…
Cancel
Save