Skip to content

Commit

Permalink
Kokkos V7 has the same results as Kokkos V3.
Browse files Browse the repository at this point in the history
  • Loading branch information
seyonglee committed Apr 30, 2024
1 parent 697512e commit 7c4f6d9
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/kokkos_src_v7/propagate-toz-test_Kokkos_v7.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -866,10 +866,11 @@ int main (int argc, char* argv[]) {
const MPHIT* bhits_ = hit.data();
MPTRK_<N> obtracks;
const auto& btracks = btracks_[tid].load_component<N>(batch_id);
obtracks = btracks;
#pragma unroll
for(size_t layer=0; layer<nlayer; ++layer) {
const auto& bhits = bhits_[layer+nlayer*tid].load_component<N>(batch_id);
propagateToZ<N>(btracks.cov, btracks.par, btracks.q, bhits.pos, obtracks.cov, obtracks.par); // vectorized function
propagateToZ<N>(obtracks.cov, obtracks.par, obtracks.q, bhits.pos, obtracks.cov, obtracks.par); // vectorized function
KalmanUpdate_v2<N>(obtracks.cov,obtracks.par,bhits.cov,bhits.pos);
}
obtracks_[tid].save_component<N>(obtracks, batch_id);
Expand Down

0 comments on commit 7c4f6d9

Please sign in to comment.