jianmu
/
lammps
forked from lijiext/lammps
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Enabling new kokkos view 

git-svn-id: svn://svn.icms.temple.edu/lammps-ro/trunk@14934 f3b2605a-c512-4ea7-a41b-209d697bcdaa
This commit is contained in:
stamoor 2016-05-03 16:23:57 +00:00
parent 955c8d92ba
commit c696881836
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/KOKKOS/kokkos_type.h
View File

@ -745,8 +745,8 @@ template<class ViewType>
void memset_kokkos (ViewType &view) {
static MemsetZeroFunctor<typename ViewType::execution_space> f;
f.ptr = view.ptr_on_device();
#ifdef KOKKOS_USING_EXPERIMENTAL_VIEW
Kokkos::parallel_for(view.memory_span()/4, f);
#ifndef KOKKOS_USING_DEPRECATED_VIEW
Kokkos::parallel_for(view.span()*sizeof(typename ViewType::value_type)/4, f);
#else
Kokkos::parallel_for(view.capacity()*sizeof(typename ViewType::value_type)/4, f);
#endif