Browse Source

!6158 [VM]Fix Print cannot export AIR model files.

Merge pull request !6158 from zjun/export_print
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
3c1e964161
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.cc
  2. +1
    -1
      mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.h

+ 1
- 1
mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.cc View File

@@ -17,13 +17,13 @@
#include "transform/graph_ir/op_declare/logging_ops_declare.h"

namespace mindspore::transform {
#ifdef ENABLE_GE
// Print
INPUT_MAP(Print) = EMPTY_INPUT_MAP;
DYN_INPUT_MAP(Print) = {{1, DYN_INPUT_DESC(x)}};
ATTR_MAP(Print) = EMPTY_ATTR_MAP;
REG_ADPT_DESC(Print, kNamePrint, ADPT_DESC(Print))

#ifdef ENABLE_GE
INPUT_MAP(Assert) = {{1, INPUT_DESC(input_condition)}};
DYN_INPUT_MAP(Assert) = {{2, DYN_INPUT_DESC(input_data)}};
ATTR_MAP(Assert) = {{"summarize", ATTR_DESC(summarize, AnyTraits<int>())}};


+ 1
- 1
mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.h View File

@@ -23,10 +23,10 @@
#include "ops/logging_ops.h"

namespace mindspore::transform {
#ifdef ENABLE_GE
DECLARE_OP_ADAPTER(Print)
DECLARE_OP_USE_DYN_INPUT(Print)

#ifdef ENABLE_GE
DECLARE_OP_ADAPTER(Assert)
DECLARE_OP_USE_DYN_INPUT(Assert)
#endif


Loading…
Cancel
Save