-
Notifications
You must be signed in to change notification settings - Fork 98
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
src/batched: Add BatchedGemm interface #935
Merged
Merged
Commits on May 27, 2021
-
src/batched: Add host-level batched GEMM interface
- Add new BatchedGemm function to the KokkosBatched namespace. This function selects how to invoke the SerialGemm routine based on the template arguments, target architecture, and Kokkos backend. The invocation selection was informed by performance data collected on V100, intel, and A64FX using the batched gemm performance tests. - Add unit tests for the BatchedGemm function.
Configuration menu - View commit details
-
Copy full SHA for 0e6204f - Browse repository at this point
Copy the full SHA 0e6204fView commit details -
Configuration menu - View commit details
-
Copy full SHA for 3932f1f - Browse repository at this point
Copy the full SHA 3932f1fView commit details -
Configuration menu - View commit details
-
Copy full SHA for ac82ecd - Browse repository at this point
Copy the full SHA ac82ecdView commit details -
src/batched: Move unused macro to common code
- Move contents of src/common/KokkosKernels_Macros.hpp to src/KokkosKernels_Macros.hpp. - Update KokkosSparse_CrsMatrix.hpp and KokkosBatched_Gemm_Decl.hpp to use unused macro.
Configuration menu - View commit details
-
Copy full SHA for 9015373 - Browse repository at this point
Copy the full SHA 9015373View commit details -
Configuration menu - View commit details
-
Copy full SHA for 8662294 - Browse repository at this point
Copy the full SHA 8662294View commit details -
Configuration menu - View commit details
-
Copy full SHA for b499019 - Browse repository at this point
Copy the full SHA b499019View commit details -
Configuration menu - View commit details
-
Copy full SHA for 67ffe9d - Browse repository at this point
Copy the full SHA 67ffe9dView commit details -
Configuration menu - View commit details
-
Copy full SHA for d821949 - Browse repository at this point
Copy the full SHA d821949View commit details
Commits on Jun 1, 2021
-
Configuration menu - View commit details
-
Copy full SHA for e45f0e4 - Browse repository at this point
Copy the full SHA e45f0e4View commit details
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.