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 "utils/comm_manager.h"
17 #include "common/common_test.h"
18
19 namespace mindspore {
20 class TestCommManager : public UT::Common {
21 public:
TestCommManager()22 TestCommManager() {}
23 };
24
TEST_F(TestCommManager,TestCreate)25 TEST_F(TestCommManager, TestCreate) {
26 std::vector<unsigned int> devices{0, 1, 2};
27 ASSERT_TRUE(CommManager::GetInstance().CreateGroupSync(string("1-2-3"), devices));
28 ASSERT_TRUE(CommManager::GetInstance().CreateGroupSync(string("hccl_world_group"), devices));
29 }
30
TEST_F(TestCommManager,TestGetSize)31 TEST_F(TestCommManager, TestGetSize) {
32 unsigned int rank_size = 0;
33 ASSERT_TRUE(CommManager::GetInstance().GetRankSize(string("1-2-3"), &rank_size));
34 ASSERT_TRUE(CommManager::GetInstance().GetRankSize(string("hccl_world_group"), &rank_size));
35 }
36
37 } // namespace mindspore
38