Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update Kokkos and Kokkos Kernels to 3.7.00 #361

Merged
merged 13 commits into from
Oct 5, 2022
Merged

Conversation

AmintorDusko
Copy link
Contributor

@AmintorDusko AmintorDusko commented Sep 23, 2022

Context: Upgrade Kokkos and Kokkos Kernels to the 3.7.00 version, and improve our sparse matrix-vector product implementation.

Description of the Change:

  • Upgrade Kokkos and Kokkos Kernels to the 3.7.00 version

  • Improve our sparse matrix-vector product implementation.

Benefits: The new implementation is faster and uses less memory.

Possible Drawbacks:

Related GitHub Issues:

@codecov
Copy link

codecov bot commented Sep 26, 2022

Codecov Report

Merging #361 (c2fc536) into master (e9fa573) will decrease coverage by 0.00%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##           master     #361      +/-   ##
==========================================
- Coverage   99.88%   99.88%   -0.01%     
==========================================
  Files          49       49              
  Lines        4406     4403       -3     
==========================================
- Hits         4401     4398       -3     
  Misses          5        5              
Impacted Files Coverage Δ
pennylane_lightning/_version.py 100.00% <100.00%> (ø)
pennylane_lightning/src/util/Kokkos_Sparse.hpp 96.87% <100.00%> (-0.27%) ⬇️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

@AmintorDusko AmintorDusko marked this pull request as ready for review September 26, 2022 13:57
@AmintorDusko AmintorDusko marked this pull request as draft September 27, 2022 12:22
@AmintorDusko AmintorDusko marked this pull request as ready for review October 3, 2022 12:36
Copy link
Member

@mlxd mlxd left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Quick query

pennylane_lightning/src/util/Kokkos_Sparse.hpp Outdated Show resolved Hide resolved
@AmintorDusko AmintorDusko marked this pull request as draft October 4, 2022 15:07
@AmintorDusko
Copy link
Contributor Author

AmintorDusko commented Oct 4, 2022

Hi @mlxd,
Here are some benchmarks showcasing the increase in performance.
Master:
image

This PR:
image

We are faster and use less memory now. 😉
Edit: The number is the number of qubits, so we are going from 20 to 26 qubits here.

@AmintorDusko AmintorDusko marked this pull request as ready for review October 4, 2022 17:10
@mlxd
Copy link
Member

mlxd commented Oct 4, 2022

Awesome result! Glad to see we can circumvent that allocation.

Copy link
Member

@mlxd mlxd left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Happy to approve!

Copy link
Contributor

@chaeyeunpark chaeyeunpark left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks @AmintorDusko! Just one question.

Copy link
Contributor

@trevor-vincent trevor-vincent left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice work Amintor, I didn't see any issues!

@AmintorDusko AmintorDusko merged commit 94d6571 into master Oct 5, 2022
@AmintorDusko AmintorDusko deleted the update-Kokkos-3.7.00 branch October 5, 2022 14:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants