-
Notifications
You must be signed in to change notification settings - Fork 407
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
Allow assignment of LayoutLeft to LayoutRight or vice versa for rank-0 Views #594
Comments
@trilinos/tpetra This will help with #957. Tpetra::Details::iallreduce used to require that both the input and the output buffers have the same array_layout. This was OK for rank-1 Views, because Kokkos allows assignment of LayoutLeft to LayoutRight or vice versa for rank-1 Views. However, Kokkos does NOT (currently) allow assignment of LayoutLeft to LayoutRight or vice versa for rank-0 Views, even though this would make sense. See the outstanding Kokkos issue: kokkos/kokkos#594 This commit adds template parameters for the array layouts of the send and receive buffers, to iallreduce. This lets the send and receive buffers have different layouts. In theory, only LayoutLeft and LayoutRight should work, though we do not currently check this. (I'm not sure if LayoutStride makes sense for a rank-0 View.)
I didn't realize this. @hcedwar The urgency comes from a deliverable for ATDM deliverable to applications by August 20. Can we prioritize this ? |
@hcedwar Let me run the tests again after kokkos commit, and I will let you know if the problem continues. |
@hcedwar |
This works for rank-1 Views, but not for rank-0 Views (where it equally would make sense to work). This is not urgent for me, but it would be helpful (e.g., would reduce the need for some template parameters in my fix for trilinos/Trilinos#957 ). Thanks!
The text was updated successfully, but these errors were encountered: