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

Update Kokkos library in LAMMPS to v3.6.0

Browse Source
This commit is contained in:
Stan Gerald Moore
2022-05-05 11:44:47 -06:00
parent bd4bbbddbe
commit b79c0bc7b4
380 changed files with 41928 additions and 8786 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lib/kokkos/benchmarks/bytes_and_flops/main.cpp
View File

@ -47,12 +47,21 @@
#include <bench.hpp>
#include <cstdlib>
extern template void run_stride_unroll<float>(int, int, int, int, int, int, int,
int);
extern template void run_stride_unroll<double>(int, int, int, int, int, int,
int, int);
extern template void run_stride_unroll<int32_t>(int, int, int, int, int, int,
int, int);
extern template void run_stride_unroll<int64_t>(int, int, int, int, int, int,
int, int);
int main(int argc, char* argv[]) {
Kokkos::initialize();
if (argc < 10) {
printf("Arguments: N K R D U F T S\n");
printf(" P: Precision (1==float, 2==double)\n");
printf(" P: Precision (1==float, 2==double, 3==int32_t, 4==int64_t)\n");
printf(" N,K: dimensions of the 2D array to allocate\n");
printf(" R: how often to loop through the K dimension with each team\n");
printf(" D: distance between loaded elements (stride)\n");
@ -91,8 +100,8 @@ int main(int argc, char* argv[]) {
printf("D must be one of 1,2,4,8,16,32\n");
return 0;
}
if ((P != 1) && (P != 2)) {
printf("P must be one of 1,2\n");
if ((P < 1) && (P > 2)) {
printf("P must be one of 1,2,3,4\n");
return 0;
}
@ -102,6 +111,12 @@ int main(int argc, char* argv[]) {
if (P == 2) {
run_stride_unroll<double>(N, K, R, D, U, F, T, S);
}
if (P == 3) {
run_stride_unroll<int32_t>(N, K, R, D, U, F, T, S);
}
if (P == 4) {
run_stride_unroll<int64_t>(N, K, R, D, U, F, T, S);
}
Kokkos::finalize();
}