|  |  | @ -32,6 +32,9 @@ | 
			
		
	
		
		
			
				
					
					|  |  |  | #include "graph/common/ge_call_wrapper.h" |  |  |  | #include "graph/common/ge_call_wrapper.h" | 
			
		
	
		
		
			
				
					
					|  |  |  | #include "register/op_registry.h" |  |  |  | #include "register/op_registry.h" | 
			
		
	
		
		
			
				
					
					|  |  |  | #include "common/ge/tbe_plugin_manager.h" |  |  |  | #include "common/ge/tbe_plugin_manager.h" | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | #ifndef ONLY_COMPILE_OPEN_SRC | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | #include "toolchain/plog.h" | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | #endif | 
			
		
	
		
		
			
				
					
					|  |  |  | 
 |  |  |  | 
 | 
			
		
	
		
		
			
				
					
					|  |  |  | using domi::OpRegistry; |  |  |  | using domi::OpRegistry; | 
			
		
	
		
		
			
				
					
					|  |  |  | using std::map; |  |  |  | using std::map; | 
			
		
	
	
		
		
			
				
					|  |  | @ -129,6 +132,11 @@ Status GEInitializeImpl(const std::map<string, string> &options) { | 
			
		
	
		
		
			
				
					
					|  |  |  | 
 |  |  |  | 
 | 
			
		
	
		
		
			
				
					
					|  |  |  | // Initialize GE, prepare for execution, call GELib::Initialize
 |  |  |  | // Initialize GE, prepare for execution, call GELib::Initialize
 | 
			
		
	
		
		
			
				
					
					|  |  |  | Status GEInitialize(const std::map<string, string> &options) { |  |  |  | Status GEInitialize(const std::map<string, string> &options) { | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | #ifndef ONLY_COMPILE_OPEN_SRC | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |   if (DlogReportInitialize() != SUCCESS) { | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |     GELOGW("Dlog report device log initialize failed."); | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |   } | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | #endif | 
			
		
	
		
		
			
				
					
					|  |  |  |   return GEInitializeImpl(options); |  |  |  |   return GEInitializeImpl(options); | 
			
		
	
		
		
			
				
					
					|  |  |  | } |  |  |  | } | 
			
		
	
		
		
			
				
					
					|  |  |  | 
 |  |  |  | 
 | 
			
		
	
	
		
		
			
				
					|  |  | @ -143,6 +151,11 @@ Status GEInitialize(const std::map<AscendString, AscendString> &options) { | 
			
		
	
		
		
			
				
					
					|  |  |  |     std::string val = option.second.GetString(); |  |  |  |     std::string val = option.second.GetString(); | 
			
		
	
		
		
			
				
					
					|  |  |  |     str_options[key] = val; |  |  |  |     str_options[key] = val; | 
			
		
	
		
		
			
				
					
					|  |  |  |   } |  |  |  |   } | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | #ifndef ONLY_COMPILE_OPEN_SRC | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |   if (DlogReportInitialize() != SUCCESS) { | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |     GELOGW("Dlog report device log initialize failed."); | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |   } | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | #endif | 
			
		
	
		
		
			
				
					
					|  |  |  |   return GEInitializeImpl(str_options); |  |  |  |   return GEInitializeImpl(str_options); | 
			
		
	
		
		
			
				
					
					|  |  |  | } |  |  |  | } | 
			
		
	
		
		
			
				
					
					|  |  |  | 
 |  |  |  | 
 | 
			
		
	
	
		
		
			
				
					|  |  | @ -187,6 +200,12 @@ Status GEFinalize() { | 
			
		
	
		
		
			
				
					
					|  |  |  |   // to avoid memory fragment, use malloc_trim to back free stack to system
 |  |  |  |   // to avoid memory fragment, use malloc_trim to back free stack to system
 | 
			
		
	
		
		
			
				
					
					|  |  |  |   malloc_trim(0); |  |  |  |   malloc_trim(0); | 
			
		
	
		
		
			
				
					
					|  |  |  | 
 |  |  |  | 
 | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | #ifndef ONLY_COMPILE_OPEN_SRC | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |   if (DlogReportFinalize() != SUCCESS) { | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |     GELOGW("Dlog report device log finalize failed."); | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  |   } | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | #endif | 
			
		
	
		
		
			
				
					
					|  |  |  |  |  |  |  | 
 | 
			
		
	
		
		
			
				
					
					|  |  |  |   GELOGT(TRACE_STOP, "GEFinalize finished"); |  |  |  |   GELOGT(TRACE_STOP, "GEFinalize finished"); | 
			
		
	
		
		
			
				
					
					|  |  |  |   return ret; |  |  |  |   return ret; | 
			
		
	
		
		
			
				
					
					|  |  |  | } |  |  |  | } | 
			
		
	
	
		
		
			
				
					|  |  | 
 |