You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

debugger.cc 46 kB

5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
5 years ago
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223
  1. /**
  2. * Copyright 2020 Huawei Technologies Co., Ltd
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. #include <dirent.h>
  17. #include <stdio.h>
  18. #include <fstream>
  19. #include <tuple>
  20. #include <vector>
  21. #include <algorithm>
  22. #include <iostream>
  23. #include <cstring>
  24. #include <utility>
  25. #include <map>
  26. #include <regex>
  27. #include "debug/debugger/debugger.h"
  28. #include "debug/data_dump/dump_json_parser.h"
  29. #include "pipeline/jit/pipeline.h"
  30. #include "backend/session/anf_runtime_algorithm.h"
  31. #include "runtime/device/kernel_runtime_manager.h"
  32. #include "runtime/device/kernel_runtime.h"
  33. #include "debug/data_dump/e2e_dump_util.h"
  34. #include "utils/config_manager.h"
  35. using debugger::Chunk;
  36. using debugger::EventReply;
  37. using debugger::GraphProto;
  38. using debugger::ModelProto;
  39. using debugger::TensorProto;
  40. using debugger::WatchCondition;
  41. using debugger::WatchCondition_Condition_inf;
  42. using debugger::WatchCondition_Condition_nan;
  43. using debugger::WatchCondition_Parameter;
  44. using debugger::WatchNode;
  45. using debugger::WatchpointHit;
  46. #define CHUNK_SIZE 1024 * 1024 * 3
  47. namespace mindspore {
  48. DebuggerPtr Debugger::debugger_ = nullptr;
  49. std::mutex Debugger::instance_lock_;
  50. static const size_t PARAMETER_OUTPUT_INDEX = 0;
  51. static const size_t VALUE_NODE_OUTPUT_INDEX = 0;
  52. Debugger::Debugger()
  53. : grpc_client_(nullptr),
  54. debug_services_(nullptr),
  55. device_id_(0),
  56. device_target_(""),
  57. num_step_(0),
  58. debugger_enabled_(false),
  59. run_level_(""),
  60. node_name_(""),
  61. cur_name_(""),
  62. training_done_(false),
  63. is_dataset_graph_(false),
  64. partial_memory_(false),
  65. last_overflow_bin_(0),
  66. initial_suspend_(true),
  67. not_dataset_graph_sum_(0),
  68. version_("") {
  69. CheckDebuggerEnabledParam();
  70. if (CheckDebuggerEnabled()) {
  71. // configure partial memory reuse
  72. partial_memory_ = CheckDebuggerPartialMemoryEnabled();
  73. auto context_ptr = MsContext::GetInstance();
  74. MS_EXCEPTION_IF_NULL(context_ptr);
  75. // switch memory reuse on or off
  76. context_ptr->set_param<bool>(MS_CTX_ENABLE_MEM_REUSE, partial_memory_);
  77. // print some message about memory reuse to user
  78. if (partial_memory_) {
  79. MS_LOG(WARNING)
  80. << "Partial Memory Reuse is enabled. Note: 1. Please only set watchpoints before running the first "
  81. "step. 2. Tensor values are only available for nodes that are watched by any watchpoint.";
  82. } else {
  83. MS_LOG(WARNING)
  84. << "Memory Reuse is disabled. Set environment variable MS_DEBUGGER_PARTIAL_MEM=1 to reduce memory "
  85. "usage for large models.";
  86. }
  87. }
  88. }
  89. void Debugger::Init(const uint32_t device_id, const std::string device_target) {
  90. // access lock for public method
  91. std::lock_guard<std::mutex> a_lock(access_lock_);
  92. // save device_id
  93. MS_LOG(INFO) << "Debugger got device_id: " << device_id;
  94. device_id_ = device_id;
  95. MS_LOG(INFO) << "Debugger got device_target: " << device_target;
  96. device_target_ = device_target;
  97. version_ = "1.1.0";
  98. }
  99. void Debugger::EnableDebugger() {
  100. // reset some of the class members
  101. num_step_ = 0;
  102. debugger_enabled_ = false;
  103. partial_memory_ = false;
  104. grpc_client_ = nullptr;
  105. debug_services_ = nullptr;
  106. // see if dump using debugger backend is enabled
  107. bool dump_enabled = CheckDebuggerDumpEnabled();
  108. MS_LOG(INFO) << "dump using debugger backend = " << dump_enabled;
  109. // check if debugger enabled
  110. debugger_enabled_ = CheckDebuggerEnabled();
  111. MS_LOG(INFO) << "debugger_enabled_ = " << debugger_enabled_;
  112. if (!debugger_enabled_ && !dump_enabled) {
  113. MS_LOG(INFO) << "Not enabling debugger. Set environment variable ENABLE_MS_DEBUGGER=1 to enable debugger.";
  114. return;
  115. }
  116. if (debugger_enabled_) {
  117. // configure grpc host
  118. const char *env_host_str = std::getenv("MS_DEBUGGER_HOST");
  119. std::string host;
  120. if (env_host_str != nullptr) {
  121. if (CheckIp(env_host_str)) {
  122. MS_LOG(INFO) << "Getenv MS_DEBUGGER_HOST: " << env_host_str;
  123. host = std::string(env_host_str);
  124. } else {
  125. debugger_enabled_ = false;
  126. MS_EXCEPTION(ValueError) << "Environment variable MS_DEBUGGER_HOST isn't a valid IP address. "
  127. "Please set environment variable MS_DEBUGGER_HOST=x.x.x.x to a valid IP";
  128. }
  129. } else {
  130. MS_LOG(INFO) << "Environment variable MS_DEBUGGER_HOST doesn't exist. Using default debugger host: localhost";
  131. host = "localhost";
  132. }
  133. // configure grpc port
  134. const char *env_port_str = std::getenv("MS_DEBUGGER_PORT");
  135. std::string port;
  136. if (env_port_str != nullptr) {
  137. if (CheckPort(env_port_str)) {
  138. MS_LOG(INFO) << "Getenv MS_DEBUGGER_PORT: " << env_port_str;
  139. port = std::string(env_port_str);
  140. } else {
  141. debugger_enabled_ = false;
  142. MS_EXCEPTION(ValueError) << "Environment variable MS_DEBUGGER_PORT is not valid. Custom port ranging from 1 to "
  143. "65535";
  144. }
  145. } else {
  146. port = "50051";
  147. if (!CheckPort(port.c_str())) {
  148. MS_EXCEPTION(ValueError) << "Default MS_DEBUGGER_PORT is not valid. Custom port ranging from 1 to 65535";
  149. }
  150. MS_LOG(INFO) << "Environment variable MS_DEBUGGER_PORT doesn't exist. Using default debugger port: 50051";
  151. }
  152. // initialize grpc client
  153. grpc_client_ = std::make_unique<GrpcClient>(host, port);
  154. }
  155. debug_services_ = std::make_unique<DebugServices>();
  156. }
  157. void Debugger::SetOpOverflowBinPath(uint32_t graph_id) {
  158. #ifdef ENABLE_D
  159. // set operation overflow info
  160. overflow_bin_path_.insert(std::pair<uint32_t, std::string>(
  161. graph_id, DumpJsonParser::GetInstance().GetOpOverflowBinPath(graph_id, device_id_)));
  162. // new overflow dump files will have a timestamp greater than last_overflow_bin_
  163. auto overflow_bin_path = overflow_bin_path_.find(graph_id)->second;
  164. DIR *d;
  165. d = opendir(overflow_bin_path.c_str());
  166. if (d != nullptr) {
  167. struct dirent *dir;
  168. while ((dir = readdir(d)) != NULL) {
  169. if (dir->d_type == DT_REG) {
  170. std::string file_path = overflow_bin_path;
  171. file_path.append(dir->d_name);
  172. std::size_t found = file_path.find_last_of(".");
  173. if (found == std::string::npos) {
  174. continue;
  175. }
  176. std::string overflow_time = file_path.substr(found + 1);
  177. if (stod(overflow_time) <= last_overflow_bin_) {
  178. MS_LOG(INFO) << "Old op overflow bin folder" << file_path;
  179. continue;
  180. }
  181. last_overflow_bin_ = stod(overflow_time);
  182. }
  183. }
  184. MS_LOG(INFO) << "last op overflow bin folder" << last_overflow_bin_;
  185. closedir(d);
  186. }
  187. #endif
  188. }
  189. void Debugger::CheckDatasetSinkMode() {
  190. if (CheckDebuggerDumpEnabled() && ConfigManager::GetInstance().dataset_mode() == DS_SINK_MODE) {
  191. MS_EXCEPTION(NotSupportError)
  192. << "e2e_dump not supported on GPU with dataset_sink_mode=True. Please set dataset_sink_mode=False";
  193. }
  194. if (CheckDebuggerEnabled() && ConfigManager::GetInstance().dataset_mode() == DS_SINK_MODE) {
  195. MS_EXCEPTION(NotSupportError)
  196. << "Debugger is not supported with dataset_sink_mode=True. Please set dataset_sink_mode=False";
  197. }
  198. }
  199. bool Debugger::CheckDebuggerDumpEnabled() {
  200. // see if dump is enabled
  201. if (device_target_ == kGPUDevice) {
  202. return device::KernelRuntime::DumpDataEnabled();
  203. }
  204. return false;
  205. }
  206. bool Debugger::CheckDebuggerEnabled() {
  207. // get env variables to configure debugger
  208. const char *env_enable_char = std::getenv("ENABLE_MS_DEBUGGER");
  209. if (env_enable_char != nullptr) {
  210. std::string env_enable_str = env_enable_char;
  211. (void)std::transform(env_enable_str.begin(), env_enable_str.end(), env_enable_str.begin(), ::tolower);
  212. if (env_enable_str == "1" || env_enable_str == "true") {
  213. return true;
  214. }
  215. }
  216. return false;
  217. }
  218. void Debugger::CheckDebuggerEnabledParam() {
  219. // check the value of env variable ENABLE_MS_DEBUGGER
  220. const char *env_enable_char = std::getenv("ENABLE_MS_DEBUGGER");
  221. if (env_enable_char != nullptr) {
  222. std::string env_enable_str = env_enable_char;
  223. (void)std::transform(env_enable_str.begin(), env_enable_str.end(), env_enable_str.begin(), ::tolower);
  224. if (env_enable_str != "0" && env_enable_str != "1" && env_enable_str != "false" && env_enable_str != "true") {
  225. MS_LOG(WARNING) << "Env variable ENABLE_MS_DEBUGGER should be True/False/1/0 (case insensitive), but get: "
  226. << env_enable_str;
  227. }
  228. }
  229. }
  230. bool Debugger::CheckDebuggerPartialMemoryEnabled() {
  231. const char *env_partial_mem_str = std::getenv("MS_DEBUGGER_PARTIAL_MEM");
  232. if (env_partial_mem_str != nullptr) {
  233. MS_LOG(INFO) << "Getenv MS_DEBUGGER_PARTIAL_MEM: " << env_partial_mem_str;
  234. if (std::strcmp(env_partial_mem_str, "1") == 0) {
  235. return true;
  236. }
  237. }
  238. return false;
  239. }
  240. bool Debugger::DebuggerBackendEnabled() { return CheckDebuggerDumpEnabled() || CheckDebuggerEnabled(); }
  241. void Debugger::Reset() {
  242. // access lock for public method
  243. std::lock_guard<std::mutex> a_lock(access_lock_);
  244. // reset components
  245. device_id_ = 0;
  246. device_target_ = "";
  247. num_step_ = 0;
  248. debugger_enabled_ = false;
  249. is_dataset_graph_ = false;
  250. partial_memory_ = false;
  251. graph_ptr_ = nullptr;
  252. grpc_client_ = nullptr;
  253. debug_services_ = nullptr;
  254. last_overflow_bin_ = 0;
  255. overflow_bin_path_.clear();
  256. stream_task_to_opname_.clear();
  257. }
  258. void Debugger::PreExecute(const KernelGraphPtr &graph_ptr, uint32_t graph_sum) {
  259. // access lock for public method
  260. std::lock_guard<std::mutex> a_lock(access_lock_);
  261. CheckDatasetSinkMode();
  262. auto graph_id = graph_ptr->graph_id();
  263. // collect rungrap_ids to update step number in multigraph case
  264. if (!rungraph_id_list_.size()) {
  265. rungraph_id_list_.push_back(graph_id);
  266. } else {
  267. if (std::find(rungraph_id_list_.begin(), rungraph_id_list_.end(), graph_id) == rungraph_id_list_.end()) {
  268. rungraph_id_list_.push_back(graph_id);
  269. }
  270. }
  271. // check and save graph_ptr, suspend if graph is new
  272. MS_LOG(INFO) << "total number graph: " << graph_sum;
  273. // multiple graphs
  274. if (graph_sum > 1) {
  275. // there are more than one graphs are not dataset_graph
  276. if (not_dataset_graph_sum_ > 0) {
  277. // only try to enable debugger if they are not all dataset graphs
  278. if (!debugger_enabled_) {
  279. EnableDebugger();
  280. }
  281. if (debugger_enabled_) {
  282. if (graph_proto_list_.size()) {
  283. // only send compiled graphs once.
  284. auto dbg_graph_ptr = graph_ptr_;
  285. // use current graph ptr to load parameters
  286. graph_ptr_ = graph_ptr;
  287. LoadParametersAndConst();
  288. // revert graph ptr to original value
  289. graph_ptr_ = dbg_graph_ptr;
  290. SendMultiGraphsAndSuspend(graph_proto_list_, graph_sum);
  291. graph_proto_list_.clear();
  292. } else if (graph_id == rungraph_id_list_.front() && device_target_ == kGPUDevice) {
  293. // stop only when receive the first sub run graph for each step
  294. CommandLoop();
  295. }
  296. }
  297. }
  298. } else if (graph_proto_list_.size() == 1) {
  299. // In single graph case, reset graph_ptr_ to be nullptr for the initial step
  300. if (num_step_ == 0) {
  301. graph_ptr_ = nullptr;
  302. }
  303. CheckGraphPtr(graph_ptr);
  304. }
  305. }
  306. void Debugger::PostExecute() {
  307. // access lock for public method
  308. std::lock_guard<std::mutex> a_lock(access_lock_);
  309. if (pipeline::ExecutorPy::GetDebugTerminate()) {
  310. return;
  311. }
  312. if (debugger_->DebuggerBackendEnabled()) {
  313. // analyze tensor data and send the watchpoints been hit
  314. if (debugger_enabled_ && !is_dataset_graph_) {
  315. if (device_target_ != kGPUDevice) {
  316. num_step_++;
  317. }
  318. MS_LOG(INFO) << "Debugger suspend at end of step; number of steps executed: " << num_step_;
  319. SendWatchpoints(CheckWatchpoints());
  320. CommandLoop();
  321. }
  322. // Only keep parameters in the current map
  323. debug_services_->ResetLoadedTensors();
  324. }
  325. }
  326. bool Debugger::ReadNodeDataRequired(const CNodePtr &kernel) {
  327. if (debugger_enabled_ && !is_dataset_graph_) {
  328. auto is_watchpoint = debug_services_->IsWatchPoint(cur_name_, kernel);
  329. // if node has a watchpoint on it, is next_to node, or continue_to node then read the kernel tensor data
  330. if (is_watchpoint || (run_level_ == "node" && (node_name_ == "" || node_name_ == cur_name_))) {
  331. return true;
  332. }
  333. }
  334. return false;
  335. }
  336. void Debugger::PostExecuteNode(const CNodePtr &kernel, bool last_kernel) {
  337. // access lock for public method
  338. std::lock_guard<std::mutex> a_lock(access_lock_);
  339. if (pipeline::ExecutorPy::GetDebugTerminate()) {
  340. return;
  341. }
  342. if (debugger_enabled_ && !is_dataset_graph_) {
  343. auto is_watchpoint = debug_services_->IsWatchPoint(cur_name_, kernel);
  344. // if kernel is watchpoint,and get hit. suspend.
  345. bool hit_empty_flag = true;
  346. if (is_watchpoint) {
  347. auto hits = CheckWatchpoints(cur_name_, kernel);
  348. if (!hits.empty()) {
  349. SendWatchpoints(hits);
  350. CommandLoop();
  351. hit_empty_flag = false;
  352. }
  353. }
  354. if (hit_empty_flag && run_level_ == "node" && (node_name_ == "" || node_name_ == cur_name_) && !last_kernel) {
  355. // if kernel is not watchpoint and is next_to or continue_to node, suspend
  356. // No need to suspend if this is the last node in graph since PostExecute suspends at the end of graph
  357. CommandLoop();
  358. }
  359. return;
  360. }
  361. }
  362. void Debugger::PostDebugOp() {
  363. // access lock for public method
  364. std::lock_guard<std::mutex> a_lock(access_lock_);
  365. // suspend if debugger is enabled
  366. if (debugger_enabled_ && !is_dataset_graph_) {
  367. MS_LOG(INFO) << "Debugger suspend at debug_op";
  368. CommandLoop();
  369. }
  370. }
  371. void Debugger::SetStreamTaskToOpnameMap(const std::map<std::pair<uint32_t, uint32_t>, std::string> &mapping) {
  372. stream_task_to_opname_ = mapping;
  373. }
  374. void Debugger::LoadGraphs(const KernelGraphPtr &graph_ptr) {
  375. if (graph_ptr_ != graph_ptr) {
  376. MS_LOG(INFO) << "LoadGraphs Debugger got new graph: " << graph_ptr->graph_id();
  377. // save new graph_ptr
  378. graph_ptr_ = graph_ptr;
  379. CheckDatasetGraph();
  380. if (!is_dataset_graph_) {
  381. // get proto for new graph_ptr
  382. auto graph_proto = GetGraphProto(graph_ptr);
  383. // add new graph proto to graph_proto_list_
  384. graph_proto_list_.push_back(graph_proto);
  385. graph_ptr_list_.push_back(graph_ptr);
  386. #ifdef ENABLE_D
  387. SetOpOverflowBinPath(graph_ptr->graph_id());
  388. #endif
  389. not_dataset_graph_sum_++;
  390. }
  391. // reset is_dataset_graph to be false
  392. is_dataset_graph_ = false;
  393. }
  394. }
  395. // In single graph cases, check single graph ptr
  396. void Debugger::CheckGraphPtr(const KernelGraphPtr &graph_ptr) {
  397. if (graph_ptr_ != graph_ptr) {
  398. MS_LOG(INFO) << "CheckGraphPtr Debugger got new graph: " << graph_ptr->graph_id();
  399. // save new graph_ptr
  400. graph_ptr_ = graph_ptr;
  401. if (!is_dataset_graph_) {
  402. // only try to enable debugger if it is not a dataset graph
  403. EnableDebugger();
  404. if (debugger_enabled_) {
  405. LoadParametersAndConst();
  406. // get graph proto and send to Mindinsight
  407. auto graph_proto = graph_proto_list_.front();
  408. SendGraphAndSuspend(graph_proto);
  409. }
  410. }
  411. }
  412. }
  413. void Debugger::CheckDatasetGraph() {
  414. // print parameter node names
  415. const auto &params = graph_ptr_->inputs();
  416. for (const auto &param : params) {
  417. MS_LOG(INFO) << "param: " << param->fullname_with_scope();
  418. }
  419. // check if there is GetNext or InitDataSetQueue node
  420. const auto &nodes = graph_ptr_->execution_order();
  421. for (const auto &node : nodes) {
  422. auto node_name = AnfAlgo::GetCNodeName(node);
  423. MS_LOG(INFO) << "node: " << node->fullname_with_scope();
  424. if (node_name == "GetNext" || node_name == "InitDataSetQueue") {
  425. MS_LOG(INFO) << "Not enabling debugger for graph " << graph_ptr_->graph_id() << ": found dataset graph node "
  426. << node_name;
  427. is_dataset_graph_ = true;
  428. return;
  429. }
  430. }
  431. is_dataset_graph_ = false;
  432. }
  433. GraphProto Debugger::GetGraphProto(const KernelGraphPtr &graph_ptr) const {
  434. // convert kernel graph to debugger modelproto
  435. ModelProto model = GetDebuggerFuncGraphProto(graph_ptr_);
  436. return model.graph();
  437. }
  438. void Debugger::SendGraphAndSuspend(const GraphProto &graph_proto) {
  439. if (SendMetadata(true)) {
  440. // send graph to Mindinsight server
  441. EventReply reply = grpc_client_->SendGraph(graph_proto);
  442. if (reply.status() != reply.OK) {
  443. MS_LOG(ERROR) << "Error: SendGraph failed";
  444. }
  445. // enter command loop, wait and process commands
  446. CommandLoop();
  447. }
  448. }
  449. bool Debugger::SendMetadata(bool version_check) {
  450. // prepare metadata
  451. std::string device_name = std::to_string(device_id_) + ":" + std::to_string(graph_ptr_->graph_id());
  452. Metadata metadata;
  453. metadata.set_device_name(device_name);
  454. metadata.set_cur_step(num_step_);
  455. metadata.set_backend(device_target_);
  456. metadata.set_cur_node(cur_name_);
  457. metadata.set_training_done(training_done_);
  458. metadata.set_ms_version(version_);
  459. MS_LOG(INFO) << "Is training done?" << training_done_;
  460. // set graph munber to not_dataset_graph_sum_
  461. metadata.set_graph_num(not_dataset_graph_sum_);
  462. EventReply reply_metadata = grpc_client_->SendMetadata(metadata);
  463. bool ret = false;
  464. if (reply_metadata.status() == reply_metadata.OK) {
  465. if (version_check) {
  466. // get type of the command in meta data reply, it should be version matched
  467. DebuggerCommand cmd = GetCommand(reply_metadata);
  468. if (cmd != DebuggerCommand::kVersionMatchedCMD) {
  469. MS_LOG(ERROR) << "MindInsight version is too old, Mindspore version is " << version_;
  470. Exit();
  471. } else {
  472. if (GetMiVersionMatched(reply_metadata)) {
  473. MS_LOG(INFO) << "MindSpore version is " << version_ << " matches MindInsight version.";
  474. ret = true;
  475. } else {
  476. MS_LOG(ERROR) << "MindSpore version " << version_ << ", did not match MindInsight version.";
  477. CommandLoop();
  478. }
  479. }
  480. } else {
  481. // version check is done before so we can just return true here
  482. ret = true;
  483. }
  484. } else {
  485. MS_LOG(ERROR) << "Error: SendMetadata failed";
  486. }
  487. return ret;
  488. }
  489. void Debugger::SendMultiGraphsAndSuspend(const std::list<GraphProto> &graph_proto_list, uint32_t graph_sum) {
  490. if (!SendMetadata(true)) {
  491. return;
  492. }
  493. // send multiple graphs to mindinght server
  494. // split graph into chunks if one graph is larger than chunk size
  495. std::list<Chunk> chunked_graph_proto_list;
  496. Chunk chunk;
  497. for (auto graph : graph_proto_list) {
  498. std::string str = graph.SerializeAsString();
  499. auto graph_size = graph.ByteSize();
  500. if (graph_size > CHUNK_SIZE) {
  501. auto sub_graph_str = grpc_client_->ChunkString(str, graph_size);
  502. for (unsigned int i = 0; i < sub_graph_str.size(); i++) {
  503. chunk.set_buffer(sub_graph_str[i]);
  504. chunked_graph_proto_list.push_back(chunk);
  505. if (i < sub_graph_str.size() - 1) {
  506. chunk.set_finished(false);
  507. } else {
  508. chunk.set_finished(true);
  509. chunked_graph_proto_list.push_back(chunk);
  510. }
  511. }
  512. } else {
  513. chunk.set_buffer(str);
  514. chunk.set_finished(true);
  515. chunked_graph_proto_list.push_back(chunk);
  516. }
  517. }
  518. EventReply reply = grpc_client_->SendMultiGraphs(chunked_graph_proto_list);
  519. if (reply.status() != reply.OK) {
  520. MS_LOG(ERROR) << "Error: SendGraph failed";
  521. }
  522. // enter command loop, wait and process commands
  523. CommandLoop();
  524. }
  525. void Debugger::CommandLoop() {
  526. // prepare metadata
  527. std::string device_name = std::to_string(device_id_) + ":" + std::to_string(graph_ptr_->graph_id());
  528. Metadata metadata;
  529. metadata.set_device_name(device_name);
  530. metadata.set_cur_step(num_step_);
  531. metadata.set_backend(device_target_);
  532. metadata.set_cur_node(cur_name_);
  533. metadata.set_training_done(training_done_);
  534. // loop exit flag
  535. bool run = false;
  536. int num_wait_fail = 0;
  537. const int max_num_wait_fail = 5;
  538. while (!run) {
  539. // wait for command
  540. EventReply reply = grpc_client_->WaitForCommand(metadata);
  541. if (reply.status() != reply.OK) {
  542. MS_LOG(ERROR) << "Error: WaitForCommand failed";
  543. num_wait_fail++;
  544. if (num_wait_fail > max_num_wait_fail) {
  545. MS_LOG(ERROR) << "Maximum number of WaitForCommand retry reached: exiting training session.";
  546. MS_LOG(ERROR) << "Failed to connect to MindInsight debugger server. Please check the config "
  547. "of debugger host and port.";
  548. Exit();
  549. run = true;
  550. } else {
  551. MS_LOG(ERROR) << "Number of consecutive WaitForCommand fail:" << num_wait_fail << "; Retry after "
  552. << num_wait_fail << "s";
  553. std::this_thread::sleep_for(std::chrono::milliseconds(1000 * num_wait_fail));
  554. }
  555. continue;
  556. }
  557. // get type of the command in reply
  558. DebuggerCommand cmd = GetCommand(reply);
  559. if (cmd == DebuggerCommand::kUnknownCMD) {
  560. MS_LOG(DEBUG) << "Debug: debugger received unknown command";
  561. continue;
  562. }
  563. MS_LOG(INFO) << "received command: ";
  564. switch (cmd) {
  565. case DebuggerCommand::kUnknownCMD:
  566. MS_LOG(INFO) << "UnknownCMD";
  567. break;
  568. case DebuggerCommand::kExitCMD:
  569. MS_LOG(INFO) << "ExitCMD";
  570. Exit();
  571. // Used for debugger termination
  572. run = true;
  573. break;
  574. case DebuggerCommand::kRunCMD:
  575. MS_LOG(INFO) << "RunCMD";
  576. if (GetRunLevel(reply) == "recheck") {
  577. MS_LOG(INFO) << "rechecking all watchpoints";
  578. SendWatchpoints(CheckWatchpoints("", nullptr, true));
  579. } else {
  580. // no longer the initial suspension.
  581. initial_suspend_ = false;
  582. // print run cmd content
  583. // get run_level and node_name
  584. run_level_ = GetRunLevel(reply);
  585. node_name_ = GetNodeName(reply);
  586. MS_LOG(INFO) << "run_level: " << run_level_;
  587. MS_LOG(INFO) << "node_name_: " << node_name_;
  588. // exit loop
  589. run = true;
  590. }
  591. break;
  592. case DebuggerCommand::kSetCMD:
  593. MS_LOG(INFO) << "SetCMD";
  594. MS_LOG(INFO) << "id: " << GetWatchpointID(reply);
  595. MS_LOG(INFO) << "delete: " << GetWatchpointDelete(reply);
  596. if (GetWatchpointDelete(reply)) {
  597. MS_LOG(INFO) << "Deleting watchpoint";
  598. RemoveWatchpoint(GetWatchpointID(reply));
  599. } else {
  600. MS_LOG(INFO) << "Setting watchpoint";
  601. MS_LOG(INFO) << "condition: " << GetWatchcondition(reply).condition();
  602. ProtoVector<WatchNode> recieved_nodes = GetWatchnodes(reply);
  603. for (const auto &node : recieved_nodes) {
  604. MS_LOG(INFO) << "node name: " << node.node_name();
  605. MS_LOG(INFO) << "node type: " << node.node_type();
  606. }
  607. ProtoVector<WatchCondition_Parameter> parameters = GetParameters(reply);
  608. for (const auto &parameter : parameters) {
  609. MS_LOG(INFO) << "parameter name: " << parameter.name();
  610. MS_LOG(INFO) << "parameter is disabled: " << parameter.disabled();
  611. MS_LOG(INFO) << "parameter value: " << parameter.value();
  612. }
  613. SetWatchpoint(GetWatchnodes(reply), GetWatchcondition(reply), GetWatchpointID(reply), GetParameters(reply));
  614. }
  615. break;
  616. case DebuggerCommand::kViewCMD: {
  617. MS_LOG(INFO) << "ViewCMD";
  618. // print view cmd content
  619. ProtoVector<TensorProto> received_tensors = GetTensors(reply);
  620. for (auto received_tensor : received_tensors) {
  621. MS_LOG(INFO) << "tensor node name: " << received_tensor.node_name();
  622. MS_LOG(INFO) << "tensor slot: " << received_tensor.slot();
  623. MS_LOG(INFO) << "tensor finished: " << std::boolalpha << received_tensor.finished() << std::noboolalpha;
  624. MS_LOG(INFO) << "tensor iter: " << received_tensor.iter();
  625. MS_LOG(INFO) << "tensor truncate: " << std::boolalpha << received_tensor.truncate() << std::noboolalpha;
  626. }
  627. MS_LOG(INFO) << "Sending tensors";
  628. std::list<TensorProto> tensors = LoadTensors(GetTensors(reply));
  629. // print view cmd reply
  630. for (auto tensor : tensors) {
  631. MS_LOG(INFO) << "tensor node name: " << tensor.node_name();
  632. MS_LOG(INFO) << "tensor slot: " << tensor.slot();
  633. MS_LOG(INFO) << "tensor finished: " << std::boolalpha << tensor.finished() << std::noboolalpha;
  634. MS_LOG(INFO) << "tensor iter: " << tensor.iter();
  635. MS_LOG(INFO) << "tensor truncate: " << std::boolalpha << tensor.truncate() << std::noboolalpha;
  636. MS_LOG(INFO) << "tensor dims: ";
  637. for (auto dim : tensor.dims()) {
  638. MS_LOG(INFO) << dim << ",";
  639. }
  640. MS_LOG(INFO) << "tensor dtype: " << tensor.data_type();
  641. }
  642. EventReply send_tensors_reply = grpc_client_->SendTensors(tensors);
  643. if (send_tensors_reply.status() != send_tensors_reply.OK) {
  644. MS_LOG(ERROR) << "Error: SendTensors failed";
  645. }
  646. } break;
  647. case DebuggerCommand::kVersionMatchedCMD:
  648. MS_LOG(ERROR) << "Received unexpected Version Matched CMD from Mindinsight.";
  649. Exit();
  650. break;
  651. default:
  652. MS_LOG(ERROR) << "Received unknown CMD from Mindinsight";
  653. Exit();
  654. break;
  655. }
  656. }
  657. }
  658. void AddTensorProtoInfo(TensorProto *tensor_item, TensorProto tensor) {
  659. tensor_item->set_node_name(tensor.node_name());
  660. tensor_item->set_slot(tensor.slot());
  661. tensor_item->set_iter(tensor.iter());
  662. tensor_item->set_truncate(tensor.truncate());
  663. tensor_item->clear_tensor_content();
  664. tensor_item->clear_data_type();
  665. tensor_item->clear_dims();
  666. }
  667. void Debugger::SetWatchpoint(const ProtoVector<WatchNode> &nodes, const WatchCondition &condition, const int32_t id,
  668. const ProtoVector<WatchCondition_Parameter> &parameters) {
  669. std::vector<std::tuple<std::string, bool>> check_node_list;
  670. std::vector<DebugServices::parameter_t> parameter_list;
  671. std::transform(nodes.begin(), nodes.end(), std::back_inserter(check_node_list),
  672. [](const WatchNode &node) -> std::tuple<std::string, bool> {
  673. return make_tuple(node.node_name(), node.node_type() == "scope");
  674. });
  675. std::transform(
  676. parameters.begin(), parameters.end(), std::back_inserter(parameter_list),
  677. [](const WatchCondition_Parameter &parameter) -> DebugServices::parameter_t {
  678. return DebugServices::parameter_t{parameter.name(), parameter.disabled(), parameter.value(), parameter.hit()};
  679. });
  680. debug_services_->AddWatchpoint(id, condition.condition(), condition.value(), check_node_list, parameter_list);
  681. }
  682. void Debugger::RemoveWatchpoint(const int32_t id) { debug_services_->RemoveWatchpoint(id); }
  683. std::list<TensorProto> Debugger::LoadTensors(const ProtoVector<TensorProto> &tensors) const {
  684. std::vector<std::string> name;
  685. std::vector<std::string> ret_name;
  686. std::vector<char *> data_ptr;
  687. std::vector<ssize_t> data_size;
  688. std::vector<TypePtr> dtype;
  689. std::vector<std::vector<int64_t>> shape;
  690. std::transform(tensors.begin(), tensors.end(), std::back_inserter(name), GetTensorFullName);
  691. // ret_name will contain tensor names that are found in TensorLoader
  692. // items in ret_name will be in the same order with tensors if found
  693. debug_services_->ReadNodesTensors(name, &ret_name, &data_ptr, &data_size, &dtype, &shape);
  694. std::list<TensorProto> tensor_list;
  695. unsigned int result_index = 0;
  696. for (auto tensor : tensors) {
  697. ssize_t size_iter = 0;
  698. if (result_index >= ret_name.size() || ret_name[result_index] != GetTensorFullName(tensor)) {
  699. TensorProto tensor_item;
  700. tensor_item.set_finished(true);
  701. AddTensorProtoInfo(&tensor_item, tensor);
  702. tensor_list.push_back(tensor_item);
  703. continue;
  704. }
  705. ssize_t tensor_size = data_size[result_index];
  706. while (size_iter < tensor_size) {
  707. ssize_t chunk_size = CHUNK_SIZE;
  708. TensorProto tensor_item;
  709. tensor_item.set_finished(false);
  710. if (tensor_size - size_iter <= CHUNK_SIZE) {
  711. chunk_size = tensor_size - size_iter;
  712. tensor_item.set_finished(true);
  713. }
  714. AddTensorProtoInfo(&tensor_item, tensor);
  715. // return empty tensor if didn't find the requested tensor
  716. tensor_item.set_tensor_content(data_ptr[result_index] + size_iter, chunk_size);
  717. tensor_item.set_data_type(GetDebuggerNumberDataType(dtype[result_index]));
  718. for (auto &elem : shape[result_index]) {
  719. tensor_item.add_dims(elem);
  720. }
  721. // add tensor to result list and increment result_index to check next item in ret_name
  722. tensor_list.push_back(tensor_item);
  723. size_iter += CHUNK_SIZE;
  724. }
  725. result_index++;
  726. }
  727. return tensor_list;
  728. }
  729. void Debugger::Exit() {
  730. // clear resource before exit
  731. // debugger will notify main thread to exit because main thread can only exit at step boundary
  732. pipeline::ExecutorPy::DebugTerminate(true);
  733. }
  734. std::list<WatchpointHit> Debugger::CheckWatchpoints(const std::string &watchnode, const CNodePtr &kernel,
  735. bool recheck) {
  736. std::vector<std::string> name;
  737. std::vector<std::string> slot;
  738. std::vector<int> condition;
  739. std::vector<unsigned int> watchpoint_id;
  740. std::vector<std::string> overflow_ops;
  741. std::vector<std::vector<DebugServices::parameter_t>> parameters;
  742. std::vector<int32_t> error_codes;
  743. #ifdef ENABLE_D
  744. overflow_ops = CheckOpOverflow();
  745. #endif
  746. std::vector<std::shared_ptr<TensorData>> tensor_list;
  747. if (watchnode.empty()) {
  748. tensor_list = debug_services_->GetTensor();
  749. } else {
  750. tensor_list = debug_services_->GetNodeTensor(kernel);
  751. }
  752. debug_services_->CheckWatchpoints(&name, &slot, &condition, &watchpoint_id, &parameters, &error_codes, overflow_ops,
  753. tensor_list, initial_suspend_, watchnode.empty(), recheck);
  754. std::list<WatchpointHit> hits;
  755. for (unsigned int i = 0; i < name.size(); i++) {
  756. WatchpointHit hit;
  757. std::vector<DebugServices::parameter_t> &parameter = parameters[i];
  758. hit.set_id(watchpoint_id[i]);
  759. hit.set_error_code(error_codes[i]);
  760. // here TensorProto act as a tensor indicator, not sending tensor content
  761. TensorProto *tensor_item = hit.mutable_tensor();
  762. tensor_item->set_node_name(name[i]);
  763. tensor_item->set_slot(slot[i]);
  764. tensor_item->set_finished(true);
  765. WatchCondition *condition_item = hit.mutable_watch_condition();
  766. condition_item->set_condition(debugger::WatchCondition_Condition(condition[i]));
  767. for (const auto &p : parameter) {
  768. auto x = condition_item->mutable_params()->Add();
  769. x->set_name(p.name);
  770. x->set_disabled(p.disabled);
  771. x->set_value(p.value);
  772. x->set_hit(p.hit);
  773. x->set_actual_value(p.actual_value);
  774. }
  775. hits.push_back(hit);
  776. }
  777. return hits;
  778. }
  779. void Debugger::SendWatchpoints(const std::list<WatchpointHit> &points) {
  780. // send info about watchpoint
  781. if (!points.empty()) {
  782. EventReply reply = grpc_client_->SendWatchpointHits(points);
  783. if (reply.status() != reply.OK) {
  784. MS_LOG(ERROR) << "Error: SendWatchpointHits failed";
  785. }
  786. }
  787. }
  788. bool Debugger::DumpTensorToFile(const std::string &tensor_name, bool trans_flag, const std::string &filepath,
  789. const std::string &host_fmt, const std::vector<int64_t> &host_shape, TypeId host_type,
  790. TypeId addr_type_id, const std::string &addr_format, size_t slot) const {
  791. return debug_services_.get()->DumpTensorToFile(tensor_name, trans_flag, filepath, host_fmt, host_shape, host_type,
  792. addr_type_id, addr_format, slot);
  793. }
  794. bool Debugger::DebugServicesIsWatchPoint(const std::string &kernel_name, const CNodePtr &kernel) const {
  795. return debug_services_.get()->IsWatchPoint(kernel_name, kernel);
  796. }
  797. void Debugger::EmptyTensor() { debug_services_.get()->EmptyTensor(); }
  798. void Debugger::SetTensorLoaderIterNum(uint32_t iter_num) { debug_services_.get()->SetTensorLoaderIterNum(iter_num); }
  799. void Debugger::EmptyPrevTensor() { debug_services_.get()->EmptyPrevTensor(); }
  800. uint32_t Debugger::GetTensorLoaderIterNum() const { return debug_services_.get()->GetTensorLoaderIterNum(); }
  801. bool Debugger::LoadNewTensor(const std::shared_ptr<TensorData> &tensor, bool keep_prev) {
  802. return debug_services_.get()->LoadNewTensor(tensor, keep_prev);
  803. }
  804. bool Debugger::debugger_enabled() const { return debugger_enabled_; }
  805. DebuggerCommand GetCommand(const EventReply &reply) {
  806. DebuggerCommand cmd = DebuggerCommand::kUnknownCMD;
  807. switch (reply.cmd_case()) {
  808. case debugger::EventReply::CmdCase::kExit:
  809. cmd = DebuggerCommand::kExitCMD;
  810. break;
  811. case debugger::EventReply::CmdCase::kRunCmd:
  812. cmd = DebuggerCommand::kRunCMD;
  813. break;
  814. case debugger::EventReply::CmdCase::kSetCmd:
  815. cmd = DebuggerCommand::kSetCMD;
  816. break;
  817. case debugger::EventReply::CmdCase::kViewCmd:
  818. cmd = DebuggerCommand::kViewCMD;
  819. break;
  820. case debugger::EventReply::CmdCase::kVersionMatched:
  821. cmd = DebuggerCommand::kVersionMatchedCMD;
  822. break;
  823. default:
  824. MS_LOG(DEBUG) << "Debug: UnknownCMD";
  825. break;
  826. }
  827. return cmd;
  828. }
  829. ProtoVector<WatchCondition_Parameter> GetParameters(const EventReply &reply) {
  830. if (!reply.has_set_cmd() || !reply.set_cmd().has_watch_condition()) {
  831. MS_LOG(ERROR) << "Error: Can not get Parameters from command. Returning default value: ProtoVector<Parameter>().";
  832. return ProtoVector<WatchCondition_Parameter>();
  833. }
  834. return reply.set_cmd().watch_condition().params();
  835. }
  836. ProtoVector<WatchNode> GetWatchnodes(const EventReply &reply) {
  837. if (!reply.has_set_cmd()) {
  838. MS_LOG(ERROR) << "Error: Not SetCMD, can not get WatchNodes. Returning default value: ProtoVector<WatchNode>().";
  839. return ProtoVector<WatchNode>();
  840. }
  841. return reply.set_cmd().watch_nodes();
  842. }
  843. std::string GetRunLevel(const EventReply &reply) {
  844. if (!reply.has_run_cmd()) {
  845. MS_LOG(ERROR) << "Error: Not RunCMD, can not get RunLevel. Returning default value: "
  846. "";
  847. return "";
  848. }
  849. return reply.run_cmd().run_level();
  850. }
  851. std::string GetNodeName(const EventReply &reply) {
  852. if (!reply.has_run_cmd()) {
  853. MS_LOG(ERROR) << "Error: Not RunCMD, can not get NodeName. Returning default value: "
  854. "";
  855. return "";
  856. }
  857. return reply.run_cmd().node_name();
  858. }
  859. WatchCondition GetWatchcondition(const EventReply &reply) {
  860. if (!reply.has_set_cmd() || !reply.set_cmd().has_watch_condition()) {
  861. MS_LOG(ERROR) << "Error: Can not get WatchCondition from command. Returning default value: WatchCondition().";
  862. return WatchCondition();
  863. }
  864. return reply.set_cmd().watch_condition();
  865. }
  866. int32_t GetWatchpointID(const EventReply &reply) {
  867. if (!reply.has_set_cmd()) {
  868. MS_LOG(ERROR) << "Error: Not SetCMD, can not get Watchpoint ID. Returning default value: 0.";
  869. return 0;
  870. }
  871. return reply.set_cmd().id();
  872. }
  873. bool GetWatchpointDelete(const EventReply &reply) {
  874. if (!reply.has_set_cmd()) {
  875. MS_LOG(ERROR) << "Error: Not SetCMD, can not get Watchpoint delete flag. Returning default value: false.";
  876. return false;
  877. }
  878. return reply.set_cmd().delete_();
  879. }
  880. ProtoVector<TensorProto> GetTensors(const EventReply &reply) {
  881. if (!reply.has_view_cmd()) {
  882. MS_LOG(ERROR) << "Error: Not ViewCMD, can not get Tensors. Returning default value: ProtoVector<TensorProto>().";
  883. return ProtoVector<TensorProto>();
  884. }
  885. return reply.view_cmd().tensors();
  886. }
  887. std::string GetTensorFullName(const TensorProto &tensor) {
  888. string node_name = tensor.node_name();
  889. if (tensor.truncate()) {
  890. // scopes in node name are seperated by '/'
  891. // use the name without scope if truncate is true
  892. std::size_t found = node_name.find_last_of("/");
  893. node_name = node_name.substr(found + 1);
  894. }
  895. return node_name + ":" + tensor.slot() + (tensor.iter() == "" ? "" : ":" + tensor.iter());
  896. }
  897. bool GetMiVersionMatched(const EventReply &reply) { return reply.version_matched(); }
  898. bool Debugger::partial_memory() { return partial_memory_; }
  899. void Debugger::SetCurNode(std::string cur_name) {
  900. // access lock for public method
  901. std::lock_guard<std::mutex> a_lock(access_lock_);
  902. cur_name_ = cur_name;
  903. }
  904. std::string Debugger::run_level() const { return run_level_; }
  905. void Debugger::SetStepNum(int32_t cur_num_step) {
  906. // access lock for public method
  907. std::lock_guard<std::mutex> a_lock(access_lock_);
  908. num_step_ = cur_num_step;
  909. }
  910. int32_t Debugger::step_num() const { return num_step_; }
  911. uint64_t BytestoInt64(const std::vector<char> &buffer) {
  912. uint64_t ret;
  913. ret = ((uint64_t)buffer[7] << 56) | ((uint64_t)buffer[6] << 48) | ((uint64_t)buffer[5] << 40) |
  914. ((uint64_t)buffer[4] << 32) | ((uint64_t)buffer[3] << 24) | ((uint64_t)buffer[2] << 16) |
  915. ((uint64_t)buffer[1] << 8) | ((uint64_t)buffer[0]);
  916. return ret;
  917. }
  918. #define BUF_SIZ 256
  919. std::vector<std::string> Debugger::CheckOpOverflow() {
  920. std::vector<double> bin_list;
  921. std::vector<std::string> op_names;
  922. for (const auto &[graph_id, overflow_bin_path] : overflow_bin_path_) {
  923. DIR *d;
  924. d = opendir(overflow_bin_path.c_str());
  925. MS_LOG(INFO) << "processing bin file path " << overflow_bin_path << ", graph id " << graph_id;
  926. if (d != nullptr) {
  927. struct dirent *dir = nullptr;
  928. while ((dir = readdir(d)) != NULL) {
  929. if (dir->d_type == DT_REG) {
  930. std::string file_path = overflow_bin_path;
  931. file_path.append(dir->d_name);
  932. std::string file_name = dir->d_name;
  933. std::size_t found = file_name.find_last_of(".");
  934. if (found == std::string::npos) {
  935. continue;
  936. }
  937. std::string overflow_time = file_name.substr(found + 1);
  938. if (stod(overflow_time) <= last_overflow_bin_) {
  939. MS_LOG(INFO) << "File already processed " << file_name;
  940. continue;
  941. }
  942. bin_list.push_back(stod(overflow_time));
  943. std::fstream infile;
  944. infile.open(file_path.c_str(), std::ios::binary | std::ios::in);
  945. if (!infile.is_open()) {
  946. MS_LOG(ERROR) << "Failed to open overflow bin file " << file_name;
  947. continue;
  948. }
  949. infile.seekg(313, std::ios::beg);
  950. std::vector<char> buffer;
  951. buffer.resize(BUF_SIZ);
  952. infile.read(buffer.data(), BUF_SIZ);
  953. uint64_t stream_id = BytestoInt64(std::vector<char>(buffer.begin() + 8, buffer.end()));
  954. uint64_t task_id = BytestoInt64(std::vector<char>(buffer.begin() + 16, buffer.end()));
  955. MS_LOG(INFO) << "Overflow stream_id " << stream_id << ", task_id " << task_id << ".";
  956. auto op = debugger_->stream_task_to_opname_.find(std::make_pair(stream_id, task_id));
  957. if (op != debugger_->stream_task_to_opname_.end()) {
  958. MS_LOG(ERROR) << "Overflow detected on node " << op->second << std::endl;
  959. op_names.push_back(op->second);
  960. } else {
  961. MS_LOG(INFO) << "No overflow is detected " << std::endl;
  962. }
  963. infile.close();
  964. }
  965. }
  966. } else {
  967. MS_LOG(INFO) << "OverFlow bin directory does not exist!";
  968. }
  969. closedir(d);
  970. }
  971. if (!op_names.empty()) {
  972. MS_LOG(ERROR) << "These operation overflows are detected " << op_names;
  973. }
  974. for (auto &i : bin_list) {
  975. if (i > last_overflow_bin_) {
  976. last_overflow_bin_ = i;
  977. }
  978. }
  979. auto iter_op_names = overflow_ops_.find(num_step_);
  980. if (iter_op_names == overflow_ops_.end()) {
  981. overflow_ops_.insert(std::pair<uint32_t, std::vector<std::string>>(num_step_, op_names));
  982. return op_names;
  983. }
  984. iter_op_names->second.insert(std::end(iter_op_names->second), std::begin(op_names), std::end(op_names));
  985. return iter_op_names->second;
  986. }
  987. void Debugger::SetTrainingDone(bool training_done) { training_done_ = training_done; }
  988. bool Debugger::CheckPort(const char *port) {
  989. char *p = const_cast<char *>(port);
  990. int num = 0;
  991. if (*p == '0' && *(p + 1) != '\0') return false;
  992. while (*p != '\0') {
  993. if (*p < '0' || *p > '9') return false;
  994. num = num * 10 + (*p) - '0';
  995. if (num < 1 || num > 65535) return false;
  996. p++;
  997. }
  998. return true;
  999. }
  1000. bool Debugger::CheckIp(const char *host) {
  1001. std::regex reg_ip(
  1002. "(25[0-4]|2[0-4][0-9]|1[0-9][0-9]|[1-9][0-9]|[1-9])"
  1003. "[.](25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9][0-9]|[0-9])"
  1004. "[.](25[0-5]|2[0-4][0-9]|1[0-9][0-9]|[1-9][0-9]|[0-9])"
  1005. "[.](25[0-4]|2[0-4][0-9]|1[0-9][0-9]|[1-9][0-9]|[1-9])");
  1006. std::smatch smat;
  1007. std::string host_str = std::string(host);
  1008. return std::regex_match(host_str, smat, reg_ip);
  1009. }
  1010. uint32_t Debugger::GetFirstRunGraphId() { return rungraph_id_list_.front(); }
  1011. void Debugger::LoadSingleAnfnode(const AnfNodePtr &anf_node, const size_t output_index) {
  1012. MS_EXCEPTION_IF_NULL(anf_node);
  1013. if (!anf_node->isa<Parameter>() && !anf_node->isa<ValueNode>()) {
  1014. return;
  1015. }
  1016. // for parameters and value nodes, set its execution order to be 0;
  1017. int exec_order = 0;
  1018. std::string node_name = anf_node->fullname_with_scope();
  1019. E2eDumpUtil::GetFileKernelName(NOT_NULL(&node_name));
  1020. // check if output adde exists, if not, return;
  1021. if (!AnfAlgo::OutputAddrExist(anf_node, output_index)) {
  1022. return;
  1023. }
  1024. auto addr = AnfAlgo::GetOutputAddr(anf_node, output_index);
  1025. MS_EXCEPTION_IF_NULL(addr);
  1026. auto type = AnfAlgo::GetOutputInferDataType(anf_node, output_index);
  1027. auto format = kOpFormat_DEFAULT;
  1028. string tensor_name = node_name + ':' + "0";
  1029. ShapeVector int_shapes;
  1030. auto shape = AnfAlgo::GetOutputDeviceShape(anf_node, output_index);
  1031. (void)std::transform(shape.begin(), shape.end(), std::back_inserter(int_shapes),
  1032. [](size_t inner_item) { return SizeToInt(inner_item); });
  1033. bool keep_prev;
  1034. if (anf_node->isa<Parameter>()) {
  1035. keep_prev = true;
  1036. debug_services_->MoveTensorCurrentToPrev(tensor_name);
  1037. } else {
  1038. keep_prev = false;
  1039. }
  1040. bool ret = addr->LoadMemToHost(tensor_name, exec_order, format, int_shapes, type, 0, keep_prev);
  1041. if (!ret) {
  1042. MS_LOG(ERROR) << "LoadMemToHost:"
  1043. << ", tensor_name:" << tensor_name << ", host_format:" << format << ".!";
  1044. }
  1045. }
  1046. void Debugger::LoadParametersAndConst() {
  1047. if (!(debugger_enabled_ || CheckDebuggerDumpEnabled())) return;
  1048. MS_EXCEPTION_IF_NULL(graph_ptr_);
  1049. // load parameters
  1050. MS_LOG(INFO) << "Start to load Parameters!";
  1051. const auto &parameters = graph_ptr_->inputs();
  1052. for (auto &item : parameters) {
  1053. LoadSingleAnfnode(item, PARAMETER_OUTPUT_INDEX);
  1054. }
  1055. // load value nodes
  1056. // get all constant avlues from the graph
  1057. MS_LOG(INFO) << "Start to load value nodes!";
  1058. const auto value_nodes = graph_ptr_->graph_value_nodes();
  1059. for (auto &item : value_nodes) {
  1060. LoadSingleAnfnode(item, VALUE_NODE_OUTPUT_INDEX);
  1061. }
  1062. }
  1063. void Debugger::LoadGraphOutputs() {
  1064. if (!(debugger_enabled() && device_target_ == kAscendDevice)) return;
  1065. MS_EXCEPTION_IF_NULL(graph_ptr_);
  1066. const auto &apply_kernels = graph_ptr_->execution_order();
  1067. // for kernels, execution order starts from 1
  1068. int exec_order = 1;
  1069. for (const auto &node : apply_kernels) {
  1070. MS_EXCEPTION_IF_NULL(node);
  1071. auto node_name = AnfAlgo::GetCNodeName(node);
  1072. std::string kernel_name = node->fullname_with_scope();
  1073. auto output_size = AnfAlgo::GetOutputTensorNum(node);
  1074. if (partial_memory_) {
  1075. if (!debug_services_->IsWatchPoint(kernel_name, node)) {
  1076. continue;
  1077. }
  1078. }
  1079. for (size_t j = 0; j < output_size; ++j) {
  1080. if (!AnfAlgo::OutputAddrExist(node, j)) {
  1081. MS_LOG(INFO) << "Cannot find output addr for slot " << j << " for " << node->fullname_with_scope();
  1082. continue;
  1083. }
  1084. auto addr = AnfAlgo::GetOutputAddr(node, j);
  1085. MS_EXCEPTION_IF_NULL(addr);
  1086. auto type = AnfAlgo::GetOutputInferDataType(node, j);
  1087. auto format = kOpFormat_DEFAULT;
  1088. string tensor_name = kernel_name + ':' + std::to_string(j);
  1089. ShapeVector int_shapes;
  1090. auto shape = AnfAlgo::GetOutputDeviceShape(node, j);
  1091. (void)std::transform(shape.begin(), shape.end(), std::back_inserter(int_shapes),
  1092. [](size_t inner_item) { return SizeToInt(inner_item); });
  1093. auto ret = addr->LoadMemToHost(tensor_name, exec_order, format, int_shapes, type, j, false);
  1094. if (!ret) {
  1095. MS_LOG(ERROR) << "LoadMemToHost:"
  1096. << ", tensor_name:" << tensor_name << ", host_format:" << format << ".!";
  1097. }
  1098. }
  1099. exec_order = exec_order + 1;
  1100. }
  1101. }
  1102. void Debugger::UpdateStepNum(const session::KernelGraph *graph) {
  1103. // update step number if we are processing the first graph (to support multigraph)
  1104. if (device_target_ == kGPUDevice && (debugger_enabled_ || device::KernelRuntime::DumpDataEnabledIteration()) &&
  1105. (graph->graph_id() == debugger_->GetFirstRunGraphId())) {
  1106. // access lock for public method
  1107. std::lock_guard<std::mutex> a_lock(access_lock_);
  1108. ++num_step_;
  1109. }
  1110. }
  1111. void Debugger::ClearCurrentData() {
  1112. if (device_target_ == kGPUDevice && (debugger_enabled_ || device::KernelRuntime::DumpDataEnabledIteration()))
  1113. debug_services_->EmptyCurrentTensor();
  1114. }
  1115. bool Debugger::TensorExistsInCurrent(std::string tensor_name) {
  1116. return debug_services_->TensorExistsInCurrent(tensor_name);
  1117. }
  1118. } // namespace mindspore