kokkos version of smaller kernel