Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add ut for debug info

This commit is contained in:
huanghui 2021-11-27 14:20:59 +08:00
parent e2b5ee71c2
commit 67a6beb1b1
1 changed files with 89 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

89
tests/ut/cpp/debug/debug_info_test.cc Normal file
View File

@ -0,0 +1,89 @@
/**
* Copyright 2021 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <iostream>
#include <memory>
#include <set>
#include "common/common_test.h"
#include "utils/info.h"
#include "utils/trace_base.h"
#include "ir/anf.h"
#include "ir/func_graph.h"
#include "frontend/operator/ops.h"
#include "base/core_ops.h"
namespace mindspore {
class TestDebugInfo : public UT::Common {
public:
TestDebugInfo() {}
};
// Feature: Debug info
// Description: Make a location
// Expectation: make a location with no error
TEST_F(TestDebugInfo, test_make_location) {
LocationPtr loc1 = std::make_shared<Location>("/home/workspace/a.py", 0, 4, 1, 8);
std::string s = loc1->ToString(kSourceLineTipDiscard);
std::string expect_str("In file /home/workspace/a.py(0)\n");
ASSERT_TRUE(s == expect_str);
}
// Feature: Debug info
// Description: Deduplicate the debug infos which have the same print
// Expectation: a set of debug infos is deduplicated
TEST_F(TestDebugInfo, test_location_dedup) {
LocationPtr loc1 = std::make_shared<Location>("file1.py", 0, 0, 0, 0);
NodeDebugInfoPtr debug_info1 = std::make_shared<NodeDebugInfo>();
debug_info1->set_location(loc1);
LocationPtr loc2 = std::make_shared<Location>("file1.py", 0, 0, 0, 0);
NodeDebugInfoPtr debug_info2 = std::make_shared<NodeDebugInfo>();
debug_info2->set_location(loc2);
LocationPtr loc3 = std::make_shared<Location>("file2.py", 0, 0, 0, 0);
NodeDebugInfoPtr debug_info3 = std::make_shared<NodeDebugInfo>();
debug_info3->set_location(loc3);
std::set<NodeDebugInfoPtr, DebugInfoCompare> fused_debug_info_set;
(void)fused_debug_info_set.emplace(debug_info1);
ASSERT_TRUE(fused_debug_info_set.size() == 1);
(void)fused_debug_info_set.emplace(debug_info2);
ASSERT_TRUE(fused_debug_info_set.size() == 1);
(void)fused_debug_info_set.emplace(debug_info3);
ASSERT_TRUE(fused_debug_info_set.size() == 2);
(void)fused_debug_info_set.emplace(debug_info3);
ASSERT_TRUE(fused_debug_info_set.size() == 2);
}
// Feature: Debug info
// Description: Test adding a fused debug info
// Expectation: success
TEST_F(TestDebugInfo, test_fused_debug_info) {
FuncGraphPtr fg = std::make_shared<FuncGraph>();
std::vector<AnfNodePtr> inputs;
CNodePtr cnode = std::make_shared<CNode>(inputs, fg);
ASSERT_TRUE(cnode->fused_debug_infos().size() == 0);
NodeDebugInfoPtr debug_info = std::make_shared<NodeDebugInfo>();
cnode->AddFusedDebugInfo(debug_info);
ASSERT_TRUE(cnode->fused_debug_infos().size() == 1);
}
} // namespace mindspore