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

[ready] kokkos impl gemm #197

Merged
merged 9 commits into from
May 5, 2022
Merged

Conversation

fnrizzi
Copy link
Contributor

@fnrizzi fnrizzi commented May 2, 2022

No description provided.

@fnrizzi fnrizzi changed the title gemm: kokkos impl [ready] gemm: kokkos impl May 2, 2022
@fnrizzi fnrizzi changed the title [ready] gemm: kokkos impl [ready] kokkos impl gemm May 2, 2022
@fnrizzi fnrizzi changed the title [ready] kokkos impl gemm [wip] kokkos impl gemm May 3, 2022
@fnrizzi
Copy link
Contributor Author

fnrizzi commented May 3, 2022

@mhoemmen i have updated this quite a bit, let me know what you think, when you a have a chance! I know you are busy :)

auto kokkos_p = to_kokkos_pointer(a.data());
using view_type = Kokkos::View<
decltype(kokkos_p)*,
typename LayoutMapper<typename mdspan_type::layout_type>::type
decltype(kokkos_p)*, typename LayoutMapper<NestedLayout>::type
Copy link
Contributor

Choose a reason for hiding this comment

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

Suggested change
decltype(kokkos_p)*, typename LayoutMapper<NestedLayout>::type
std::add_pointer_t<decltype(kokkos_p)>, typename LayoutMapper<NestedLayout>::type

(This converts, e.g., int* to int**, and const int* to const int**.)

Copy link
Contributor Author

@fnrizzi fnrizzi May 5, 2022

Choose a reason for hiding this comment

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

why is this wrong? this is for a rank-2 mdspan, so it becomes a rank-2 view

Copy link
Contributor

Choose a reason for hiding this comment

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

@fnrizzi It's not wrong, just a suggestion : - )

Copy link
Contributor Author

Choose a reason for hiding this comment

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

ok! suggestion taken!

@mhoemmen
Copy link
Contributor

mhoemmen commented May 4, 2022

@fnrizzi To clarify: It's fine to merge now, but you might like to fix the typos first. If you're in a hurry, just let me know and I'll merge it.

@fnrizzi
Copy link
Contributor Author

fnrizzi commented May 4, 2022

@mhoemmen no hurry, i will fix this properly!

@fnrizzi fnrizzi changed the title [wip] kokkos impl gemm [ready] kokkos impl gemm May 5, 2022
auto kokkos_p = to_kokkos_pointer(a.data());
using view_type = Kokkos::View<
decltype(kokkos_p)*,
typename LayoutMapper<typename mdspan_type::layout_type>::type
decltype(kokkos_p)*, typename LayoutMapper<NestedLayout>::type
Copy link
Contributor

Choose a reason for hiding this comment

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

@fnrizzi It's not wrong, just a suggestion : - )

@mhoemmen mhoemmen merged commit c583682 into kokkos:main May 5, 2022
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.

2 participants