-
-
Notifications
You must be signed in to change notification settings - Fork 442
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Fixing handling of bool value type for collective operations
- Loading branch information
Showing
5 changed files
with
175 additions
and
8 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
157 changes: 157 additions & 0 deletions
157
libs/full/collectives/tests/regressions/reduce_vector_bool.cpp
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,157 @@ | ||
| // Copyright (c) 2019-2024 Hartmut Kaiser | ||
| // | ||
| // SPDX-License-Identifier: BSL-1.0 | ||
| // Distributed under the Boost Software License, Version 1.0. (See accompanying | ||
| // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) | ||
|
|
||
| #include <hpx/config.hpp> | ||
|
|
||
| #if !defined(HPX_COMPUTE_DEVICE_CODE) | ||
| #include <hpx/hpx.hpp> | ||
| #include <hpx/hpx_init.hpp> | ||
| #include <hpx/modules/collectives.hpp> | ||
| #include <hpx/modules/testing.hpp> | ||
|
|
||
| #include <cstdint> | ||
| #include <iostream> | ||
| #include <string> | ||
| #include <utility> | ||
| #include <vector> | ||
|
|
||
| using namespace hpx::collectives; | ||
|
|
||
| constexpr char const* reduce_direct_basename = "/test/reduce_direct/"; | ||
| #if defined(HPX_DEBUG) | ||
| constexpr int ITERATIONS = 100; | ||
| #else | ||
| constexpr int ITERATIONS = 1000; | ||
| #endif | ||
|
|
||
| void test_multiple_use_with_generation() | ||
| { | ||
| std::uint32_t const this_locality = hpx::get_locality_id(); | ||
| std::uint32_t const num_localities = | ||
| hpx::get_num_localities(hpx::launch::sync); | ||
| HPX_TEST_LTE(static_cast<std::uint32_t>(2), num_localities); | ||
|
|
||
| auto const reduce_direct_client = | ||
| create_communicator(reduce_direct_basename, | ||
| num_sites_arg(num_localities), this_site_arg(this_locality)); | ||
|
|
||
| hpx::chrono::high_resolution_timer const t; | ||
|
|
||
| for (int i = 0; i != ITERATIONS; ++i) | ||
| { | ||
| bool value = ((this_locality + i) % 2) ? true : false; | ||
| if (this_locality == 0) | ||
| { | ||
| hpx::future<bool> overall_result = | ||
| reduce_here(reduce_direct_client, std::move(value), | ||
| std::logical_or<>{}, generation_arg(i + 1)); | ||
|
|
||
| bool sum = false; | ||
| for (std::uint32_t j = 0; j != num_localities; ++j) | ||
| { | ||
| sum = sum || (((j + i) % 2) ? true : false); | ||
| } | ||
| HPX_TEST_EQ(sum, overall_result.get()); | ||
| } | ||
| else | ||
| { | ||
| hpx::future<void> overall_result = reduce_there( | ||
| reduce_direct_client, std::move(value), generation_arg(i + 1)); | ||
| overall_result.get(); | ||
| } | ||
| } | ||
|
|
||
| auto const elapsed = t.elapsed(); | ||
| if (this_locality == 0) | ||
| { | ||
| std::cout << "remote timing: " << elapsed / ITERATIONS << "[s]\n"; | ||
| } | ||
| } | ||
|
|
||
| void test_local_use() | ||
| { | ||
| constexpr std::uint32_t num_sites = 10; | ||
|
|
||
| std::vector<hpx::future<void>> sites; | ||
| sites.reserve(num_sites); | ||
|
|
||
| // launch num_sites threads to represent different sites | ||
| for (std::uint32_t site = 0; site != num_sites; ++site) | ||
| { | ||
| sites.push_back(hpx::async([=]() { | ||
| auto const reduce_direct_client = | ||
| create_communicator(reduce_direct_basename, | ||
| num_sites_arg(num_sites), this_site_arg(site)); | ||
|
|
||
| hpx::chrono::high_resolution_timer const t; | ||
|
|
||
| // test functionality based on immediate local result value | ||
| for (int i = 0; i != ITERATIONS; ++i) | ||
| { | ||
| bool value = ((site + i) % 2) ? true : false; | ||
| if (site == 0) | ||
| { | ||
| hpx::future<bool> overall_result = reduce_here( | ||
| reduce_direct_client, std::move(value), std::logical_or<>{}, | ||
| generation_arg(i + 1), this_site_arg(site)); | ||
|
|
||
| bool sum = false; | ||
| for (std::uint32_t j = 0; j != num_sites; ++j) | ||
| { | ||
| sum = sum || (((j + i) % 2) ? true : false); | ||
| } | ||
| HPX_TEST_EQ(sum, overall_result.get()); | ||
| } | ||
| else | ||
| { | ||
| hpx::future<void> overall_result = | ||
| reduce_there(reduce_direct_client, std::move(value), | ||
| generation_arg(i + 1), this_site_arg(site)); | ||
| overall_result.get(); | ||
| } | ||
| } | ||
|
|
||
| auto const elapsed = t.elapsed(); | ||
| if (site == 0) | ||
| { | ||
| std::cout << "local timing: " << elapsed / (10 * ITERATIONS) | ||
| << "[s]\n"; | ||
| } | ||
| })); | ||
| } | ||
|
|
||
| hpx::wait_all(std::move(sites)); | ||
| } | ||
|
|
||
| int hpx_main() | ||
| { | ||
| #if defined(HPX_HAVE_NETWORKING) | ||
| if (hpx::get_num_localities(hpx::launch::sync) > 1) | ||
| { | ||
| test_multiple_use_with_generation(); | ||
| } | ||
| #endif | ||
|
|
||
| if (hpx::get_locality_id() == 0) | ||
| { | ||
| test_local_use(); | ||
| } | ||
|
|
||
| return hpx::finalize(); | ||
| } | ||
|
|
||
| int main(int argc, char* argv[]) | ||
| { | ||
| std::vector<std::string> const cfg = {"hpx.run_hpx_main!=1"}; | ||
|
|
||
| hpx::init_params init_args; | ||
| init_args.cfg = cfg; | ||
|
|
||
| HPX_TEST_EQ(hpx::init(argc, argv, init_args), 0); | ||
| return hpx::util::report_errors(); | ||
| } | ||
|
|
||
| #endif |