From 63225acf8ba99bdd1afb1002c36023fcc1442d96 Mon Sep 17 00:00:00 2001 From: caifubi Date: Wed, 2 Dec 2020 16:17:47 +0800 Subject: [PATCH] OpDebug adapt to new Ascend run package --- mindspore/ccsrc/backend/kernel_compiler/ascend_kernel_mod.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/ascend_kernel_mod.h b/mindspore/ccsrc/backend/kernel_compiler/ascend_kernel_mod.h index e847881b68..47cdd7dfff 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/ascend_kernel_mod.h +++ b/mindspore/ccsrc/backend/kernel_compiler/ascend_kernel_mod.h @@ -33,7 +33,8 @@ class AscendKernelMod : public KernelMod { uint32_t block_dim() { return block_dim_; } uint32_t stream_id() { return stream_id_; } virtual bool NeedDump() { - return DumpJsonParser::GetInstance().NeedDump(kernel_name_) && DumpJsonParser::GetInstance().async_dump_enabled(); + const auto &dump_json = DumpJsonParser::GetInstance(); + return dump_json.NeedDump(kernel_name_) && dump_json.async_dump_enabled() && dump_json.op_debug_mode() == 0; } protected: