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