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

Fix AVX2 simd support for ZEN2 AMD CPU. #6238

Merged
merged 3 commits into from
Jun 26, 2023
Merged

Conversation

pkestene
Copy link
Contributor

@pkestene pkestene commented Jun 23, 2023

Enable AVX2 vectorization when KOKKOS_ARCH_AMD_AVX2 is ON for AMD Zen architectures.
Drop KOKKOS_ARCH_AMD_AVX2 macro in favor of KOKKOS_ARCH_AVX2.

@dalg24-jenkins
Copy link
Collaborator

Can one of the admins verify this patch?

simd/src/Kokkos_SIMD.hpp Outdated Show resolved Hide resolved
@dalg24 dalg24 requested review from ldh4 and masterleinad June 23, 2023 11:53
Co-authored-by: Damien L-G <dalg24+github@gmail.com>
@Rombur
Copy link
Member

Rombur commented Jun 23, 2023

Is there a reason why we don't use KOKKOS_ARCH_AVX2 on AMD CPU?

@dalg24
Copy link
Member

dalg24 commented Jun 26, 2023

Is there a reason why we don't use KOKKOS_ARCH_AVX2 on AMD CPU?

We could not think of a good reason. We will drop KOKKOS_ARCH_AMD_AVX2 in favor of KOKKOS_ARCH_AVX2.
@Rombur agreed to implement these changes. Make sure you update the description accordingly and add to the changelog.

@dalg24 dalg24 requested review from masterleinad and ldh4 June 26, 2023 15:25
@dalg24
Copy link
Member

dalg24 commented Jun 26, 2023

Retest this please

@dalg24 dalg24 merged commit 91b57f1 into kokkos:develop Jun 26, 2023
28 checks passed
@Rombur Rombur mentioned this pull request Jun 16, 2023
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.

None yet

6 participants