ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update Kokkos library in LAMMPS to v4.2

Browse Source
This commit is contained in:
Stan Gerald Moore
2023-11-21 15:02:12 -07:00
parent 33dcfb5390
commit 3306b95589
493 changed files with 41548 additions and 15629 deletions
Download Patch File Download Diff File Expand all files Collapse all files

131
lib/kokkos/simd/unit_tests/include/TestSIMD_Conversions.hpp Normal file
View File

@ -0,0 +1,131 @@
//@HEADER
// ************************************************************************
//
// Kokkos v. 4.0
// Copyright (2022) National Technology & Engineering
// Solutions of Sandia, LLC (NTESS).
//
// Under the terms of Contract DE-NA0003525 with NTESS,
// the U.S. Government retains certain rights in this software.
//
// Part of Kokkos, under the Apache License v2.0 with LLVM Exceptions.
// See https://kokkos.org/LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//@HEADER
#ifndef KOKKOS_TEST_SIMD_CONVERSIONS_HPP
#define KOKKOS_TEST_SIMD_CONVERSIONS_HPP
#include <Kokkos_SIMD.hpp>
#include <SIMDTesting_Utilities.hpp>
template <typename Abi>
inline void host_check_conversions() {
{
auto a = Kokkos::Experimental::simd<std::uint64_t, Abi>(1);
auto b = Kokkos::Experimental::simd<std::int64_t, Abi>(a);
EXPECT_TRUE(all_of(b == decltype(b)(1)));
}
{
auto a = Kokkos::Experimental::simd<std::int32_t, Abi>(1);
auto b = Kokkos::Experimental::simd<std::uint64_t, Abi>(a);
EXPECT_TRUE(all_of(b == decltype(b)(1)));
}
{
auto a = Kokkos::Experimental::simd<std::uint64_t, Abi>(1);
auto b = Kokkos::Experimental::simd<std::int32_t, Abi>(a);
EXPECT_TRUE(all_of(b == decltype(b)(1)));
}
{
auto a = Kokkos::Experimental::simd_mask<double, Abi>(true);
auto b = Kokkos::Experimental::simd_mask<std::int32_t, Abi>(a);
EXPECT_TRUE(b == decltype(b)(true));
}
{
auto a = Kokkos::Experimental::simd_mask<std::int32_t, Abi>(true);
auto b = Kokkos::Experimental::simd_mask<std::uint64_t, Abi>(a);
EXPECT_TRUE(b == decltype(b)(true));
}
{
auto a = Kokkos::Experimental::simd_mask<std::int32_t, Abi>(true);
auto b = Kokkos::Experimental::simd_mask<std::int64_t, Abi>(a);
EXPECT_TRUE(b == decltype(b)(true));
}
{
auto a = Kokkos::Experimental::simd_mask<std::int32_t, Abi>(true);
auto b = Kokkos::Experimental::simd_mask<double, Abi>(a);
EXPECT_TRUE(b == decltype(b)(true));
}
}
template <typename... Abis>
inline void host_check_conversions_all_abis(
Kokkos::Experimental::Impl::abi_set<Abis...>) {
(host_check_conversions<Abis>(), ...);
}
template <typename Abi>
KOKKOS_INLINE_FUNCTION void device_check_conversions() {
kokkos_checker checker;
{
auto a = Kokkos::Experimental::simd<std::uint64_t, Abi>(1);
auto b = Kokkos::Experimental::simd<std::int64_t, Abi>(a);
checker.truth(all_of(b == decltype(b)(1)));
}
{
auto a = Kokkos::Experimental::simd<std::int32_t, Abi>(1);
auto b = Kokkos::Experimental::simd<std::uint64_t, Abi>(a);
checker.truth(all_of(b == decltype(b)(1)));
}
{
auto a = Kokkos::Experimental::simd<std::uint64_t, Abi>(1);
auto b = Kokkos::Experimental::simd<std::int32_t, Abi>(a);
checker.truth(all_of(b == decltype(b)(1)));
}
{
auto a = Kokkos::Experimental::simd_mask<double, Abi>(true);
auto b = Kokkos::Experimental::simd_mask<std::int32_t, Abi>(a);
checker.truth(b == decltype(b)(true));
}
{
auto a = Kokkos::Experimental::simd_mask<std::int32_t, Abi>(true);
auto b = Kokkos::Experimental::simd_mask<std::uint64_t, Abi>(a);
checker.truth(b == decltype(b)(true));
}
{
auto a = Kokkos::Experimental::simd_mask<std::int32_t, Abi>(true);
auto b = Kokkos::Experimental::simd_mask<std::int64_t, Abi>(a);
checker.truth(b == decltype(b)(true));
}
{
auto a = Kokkos::Experimental::simd_mask<std::int32_t, Abi>(true);
auto b = Kokkos::Experimental::simd_mask<double, Abi>(a);
checker.truth(b == decltype(b)(true));
}
}
template <typename... Abis>
KOKKOS_INLINE_FUNCTION void device_check_conversions_all_abis(
Kokkos::Experimental::Impl::abi_set<Abis...>) {
(device_check_conversions<Abis>(), ...);
}
class simd_device_conversions_functor {
public:
KOKKOS_INLINE_FUNCTION void operator()(int) const {
device_check_conversions_all_abis(
Kokkos::Experimental::Impl::device_abi_set());
}
};
TEST(simd, host_conversions) {
host_check_conversions_all_abis(Kokkos::Experimental::Impl::host_abi_set());
}
TEST(simd, device_conversions) {
Kokkos::parallel_for(Kokkos::RangePolicy<Kokkos::IndexType<int>>(0, 1),
simd_device_conversions_functor());
}
#endif