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.1

Browse Source
This commit is contained in:
Stan Gerald Moore
2024-03-06 14:58:06 -07:00
parent 9d9dbc1fa8
commit 6cc9a4c7b7
32 changed files with 121 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lib/kokkos/simd/src/Kokkos_SIMD_Scalar.hpp
View File

@ -224,7 +224,7 @@ template <typename T>
using data_type = std::conditional_t<std::is_floating_point_v<T>, T, double>;
return Experimental::simd<data_type, Experimental::simd_abi::scalar>(
Kokkos::floor(static_cast<data_type>(a[0])));
};
}
template <typename T>
[[nodiscard]] KOKKOS_FORCEINLINE_FUNCTION auto ceil(
@ -232,7 +232,7 @@ template <typename T>
using data_type = std::conditional_t<std::is_floating_point_v<T>, T, double>;
return Experimental::simd<data_type, Experimental::simd_abi::scalar>(
Kokkos::ceil(static_cast<data_type>(a[0])));
};
}
template <typename T>
[[nodiscard]] KOKKOS_FORCEINLINE_FUNCTION auto round(
@ -240,7 +240,7 @@ template <typename T>
using data_type = std::conditional_t<std::is_floating_point_v<T>, T, double>;
return Experimental::simd<data_type, Experimental::simd_abi::scalar>(
Experimental::round_half_to_nearest_even(static_cast<data_type>(a[0])));
};
}
template <typename T>
[[nodiscard]] KOKKOS_FORCEINLINE_FUNCTION auto trunc(
@ -248,7 +248,7 @@ template <typename T>
using data_type = std::conditional_t<std::is_floating_point_v<T>, T, double>;
return Experimental::simd<data_type, Experimental::simd_abi::scalar>(
Kokkos::trunc(static_cast<data_type>(a[0])));
};
}
template <class T>
[[nodiscard]] KOKKOS_FORCEINLINE_FUNCTION

2
lib/kokkos/simd/unit_tests/include/TestSIMD_GeneratorCtors.hpp
View File

@ -42,6 +42,7 @@ inline void host_check_gen_ctor() {
simd_type blend;
blend.copy_from(expected, Kokkos::Experimental::element_aligned_tag());
#if !(defined(KOKKOS_ENABLE_CUDA) && defined(KOKKOS_COMPILER_MSVC))
if constexpr (std::is_same_v<Abi, Kokkos::Experimental::simd_abi::scalar>) {
simd_type basic(KOKKOS_LAMBDA(std::size_t i) { return init[i]; });
host_check_equality(basic, rhs, lanes);
@ -63,6 +64,7 @@ inline void host_check_gen_ctor() {
host_check_equality(blend, result, lanes);
}
#endif
}
template <typename Abi, typename... DataTypes>