77TEST (BroadcastedVector, BroadcastData) {
88 std::size_t n = 100 ;
99 auto rank = dr::mp::default_comm ().rank ();
10- std::vector<int > data (n);
10+ std::vector<int > data (n);
1111 if (rank == 0 ) {
1212 for (int i = 0 ; i < n; i++) {
13- data[i] = i;
13+ data[i] = i;
1414 }
1515 }
1616 dr::mp::broadcasted_vector<int > broadcasted;
1717 if (rank == 0 ) {
18- broadcasted.broadcast_data (n, 0 , data,
19- dr::mp::default_comm ());
20- }
21- else {
18+ broadcasted.broadcast_data (n, 0 , data, dr::mp::default_comm ());
19+ } else {
2220 broadcasted.broadcast_data (n, 0 , rng::empty_view<int >(),
23- dr::mp::default_comm ());
21+ dr::mp::default_comm ());
2422 }
25-
23+
2624 std::vector<int > ref (n);
2725 for (int i = 0 ; i < n; i++) {
2826 ref[i] = i;
2927 }
3028
31- EXPECT_EQ (rng::subrange (broadcasted.broadcasted_data (), broadcasted.broadcasted_data () + n), ref);
29+ EXPECT_EQ (rng::subrange (broadcasted.broadcasted_data (),
30+ broadcasted.broadcasted_data () + n),
31+ ref);
3232 broadcasted.destroy_data ();
3333}
3434
@@ -38,35 +38,35 @@ TEST(BroadcastedVector, BroadcastDataReuse) {
3838 std::vector<int > data (n);
3939 if (rank == 0 ) {
4040 for (int i = 0 ; i < n; i++) {
41- data[i] = i;
41+ data[i] = i;
4242 }
4343 }
4444 dr::mp::broadcasted_vector<int > broadcasted;
4545 if (rank == 0 ) {
46- broadcasted.broadcast_data (n, 0 , data,
47- dr::mp::default_comm ());
48- }
49- else {
46+ broadcasted.broadcast_data (n, 0 , data, dr::mp::default_comm ());
47+ } else {
5048 broadcasted.broadcast_data (n, 0 , rng::empty_view<int >(),
51- dr::mp::default_comm ());
49+ dr::mp::default_comm ());
5250 }
53-
51+
5452 std::vector<int > ref (n);
5553 for (int i = 0 ; i < n; i++) {
5654 ref[i] = i;
5755 }
5856
59- EXPECT_EQ (rng::subrange (broadcasted.broadcasted_data (), broadcasted.broadcasted_data () + n), ref);
57+ EXPECT_EQ (rng::subrange (broadcasted.broadcasted_data (),
58+ broadcasted.broadcasted_data () + n),
59+ ref);
6060 broadcasted.destroy_data ();
6161 EXPECT_EQ (broadcasted.broadcasted_data (), nullptr );
62- if (rank == 0 ) {
63- broadcasted.broadcast_data (n, 0 , data,
64- dr::mp::default_comm ());
65- }
66- else {
62+ if (rank == 0 ) {
63+ broadcasted.broadcast_data (n, 0 , data, dr::mp::default_comm ());
64+ } else {
6765 broadcasted.broadcast_data (n, 0 , rng::empty_view<int >(),
68- dr::mp::default_comm ());
66+ dr::mp::default_comm ());
6967 }
70- EXPECT_EQ (rng::subrange (broadcasted.broadcasted_data (), broadcasted.broadcasted_data () + n), ref);
68+ EXPECT_EQ (rng::subrange (broadcasted.broadcasted_data (),
69+ broadcasted.broadcasted_data () + n),
70+ ref);
7171 broadcasted.destroy_data ();
72- }
72+ }
0 commit comments