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

cleanup format

Browse Source
This commit is contained in:
Axel Kohlmeyer
2025-04-06 11:16:58 -04:00
parent 2a1fb835ad
commit 911b3eb811
1 changed files with 7 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/KOKKOS/memory_kokkos.h
View File

@ -1,4 +1,3 @@
// clang-format off
/* -*- c++ -*- ---------------------------------------------------------- /* -*- c++ -*- ----------------------------------------------------------
LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator LAMMPS - Large-scale Atomic/Molecular Massively Parallel Simulator
https://www.lammps.org/, Sandia National Laboratories https://www.lammps.org/, Sandia National Laboratories
@ -11,6 +10,7 @@
See the README file in the top-level LAMMPS directory. See the README file in the top-level LAMMPS directory.
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
// clang-format off
#ifndef LMP_MEMORY_KOKKOS_H #ifndef LMP_MEMORY_KOKKOS_H
#define LMP_MEMORY_KOKKOS_H #define LMP_MEMORY_KOKKOS_H
@ -35,8 +35,7 @@ class MemoryKokkos : public Memory {
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
template <typename TYPE> template <typename TYPE>
TYPE create_kokkos(TYPE &data, typename TYPE::value_type *&array, TYPE create_kokkos(TYPE &data, typename TYPE::value_type *&array, int n1, const char *name)
int n1, const char *name)
{ {
data = TYPE(name,n1); data = TYPE(name,n1);
array = data.h_view.data(); array = data.h_view.data();
@ -44,8 +43,7 @@ TYPE create_kokkos(TYPE &data, typename TYPE::value_type *&array,
} }
template <typename TYPE, typename HTYPE> template <typename TYPE, typename HTYPE>
TYPE create_kokkos(TYPE &data, HTYPE &h_data, TYPE create_kokkos(TYPE &data, HTYPE &h_data, typename TYPE::value_type *&array, int n1,
typename TYPE::value_type *&array, int n1,
const char *name) const char *name)
{ {
data = TYPE(std::string(name),n1); data = TYPE(std::string(name),n1);
@ -56,8 +54,7 @@ template <typename TYPE, typename HTYPE>
template <typename TYPE, typename HTYPE> template <typename TYPE, typename HTYPE>
TYPE create_kokkos(TYPE &data, HTYPE &h_data, TYPE create_kokkos(TYPE &data, HTYPE &h_data, int n1, const char *name)
int n1, const char *name)
{ {
data = TYPE(std::string(name),n1); data = TYPE(std::string(name),n1);
h_data = Kokkos::create_mirror_view(data); h_data = Kokkos::create_mirror_view(data);
@ -69,8 +66,7 @@ template <typename TYPE, typename HTYPE>
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
template <typename TYPE> template <typename TYPE>
TYPE grow_kokkos(TYPE &data, typename TYPE::value_type *&array, TYPE grow_kokkos(TYPE &data, typename TYPE::value_type *&array, int n1, const char *name)
int n1, const char *name)
{ {
if (array == nullptr) return create_kokkos(data,array,n1,name); if (array == nullptr) return create_kokkos(data,array,n1,name);
@ -96,12 +92,11 @@ void destroy_kokkos(TYPE data, typename TYPE::value_type* &array)
------------------------------------------------------------------------- */ ------------------------------------------------------------------------- */
template <typename TYPE, typename HTYPE> template <typename TYPE, typename HTYPE>
TYPE create_kokkos(TYPE &data, HTYPE &h_data, int n1, int n2, TYPE create_kokkos(TYPE &data, HTYPE &h_data, int n1, int n2, const char *name)
const char *name)
{ {
data = TYPE(std::string(name),n1,n2); data = TYPE(std::string(name),n1,n2);
h_data = Kokkos::create_mirror_view(data); h_data = Kokkos::create_mirror_view(data);
//printf(">>> name: %s\n", name);
return data; return data;
} }
@ -112,7 +107,6 @@ TYPE create_kokkos(TYPE &data, typename TYPE::value_type **&array,
data = TYPE(std::string(name),n1,n2); data = TYPE(std::string(name),n1,n2);
bigint nbytes = ((bigint) sizeof(typename TYPE::value_type *)) * n1; bigint nbytes = ((bigint) sizeof(typename TYPE::value_type *)) * n1;
array = (typename TYPE::value_type **) smalloc(nbytes,name); array = (typename TYPE::value_type **) smalloc(nbytes,name);
//printf(">>> name %s nbytes %d\n", name, nbytes);
for (int i = 0; i < n1; i++) { for (int i = 0; i < n1; i++) {
if (n2 == 0) if (n2 == 0)