forked from huawei/mindspore2022
90 lines
3.0 KiB
C++
90 lines
3.0 KiB
C++
/**
|
|
* 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
|