From 5bc855416caf6c6eb51bd0a0c3a4cf7c524342ef Mon Sep 17 00:00:00 2001 From: zjun Date: Mon, 14 Sep 2020 11:40:47 +0800 Subject: [PATCH] fix export print AIR model files Signed-off-by: zjun fix graphengine commit Signed-off-by: zjun --- .../ccsrc/transform/graph_ir/op_declare/logging_ops_declare.cc | 2 +- .../ccsrc/transform/graph_ir/op_declare/logging_ops_declare.h | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.cc b/mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.cc index 8b7757f85d..084c3572b2 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.cc +++ b/mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.cc @@ -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())}}; diff --git a/mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.h b/mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.h index 5419edf18a..3110e96cc2 100644 --- a/mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.h +++ b/mindspore/ccsrc/transform/graph_ir/op_declare/logging_ops_declare.h @@ -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