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

Implement rocSOLVER dsyevd for MAGMA build #597

Merged
merged 1 commit into from
Mar 7, 2022
Merged

Conversation

mewall
Copy link
Collaborator

@mewall mewall commented Mar 7, 2022

o Pattern after cuSOLVER implementation
o Build is controlled using BML_ROCSOLVER
- Like BML_CUSOLVER

Copy link
Collaborator

@nicolasbock nicolasbock left a comment

Choose a reason for hiding this comment

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

LGTM

(I don't have the necessary GPU hardware though to test this.)

CMakeLists.txt Outdated
@@ -253,6 +253,35 @@ if (CUDA_FOUND)
include_directories(${CUDA_INCLUDE_DIRS})
endif()

list(APPEND CMAKE_PREFIX_PATH $ENV{ROCM_PATH})
Copy link
Collaborator

Choose a reason for hiding this comment

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

Shouldn't it be done in the build script instead of here?

CMakeLists.txt Outdated
endif()

if(BML_ROCSOLVER)
find_package(rocblas)
Copy link
Collaborator

Choose a reason for hiding this comment

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

Should REQUIRED be added as an argument here? Build should fail if the roc solver is requested but rocblas is not found, right?

CMakeLists.txt Outdated
@@ -454,6 +492,9 @@ if(MAGMA_FOUND)
if(BML_CUSOLVER)
list(APPEND LINK_LIBRARIES ${CUDA_cusolver_LIBRARY})
endif()
if(BML_ROCSOLVER)
list(APPEND LINK_LIBRARIES ${ROCM_rocsolver_LIBRARY})
Copy link
Collaborator

Choose a reason for hiding this comment

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

rocsolver_LIBRARIES is used earlier. Is it the same as ROCM_ro?csolver_LIBRARIES

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

I just pushed a revision. Are these comments addressed?

@jeanlucf22
Copy link
Collaborator

@mewall Did you verify that none of your additions conflict with the magma build on NVIDIA?

  o Pattern after cuSOLVER implementation
  o Build is controlled using BML_ROCSOLVER
    - Like BML_CUSOLVER
@nicolasbock nicolasbock merged commit a2578d4 into master Mar 7, 2022
@nicolasbock nicolasbock deleted the rocsolver branch March 7, 2022 17:49
@mewall
Copy link
Collaborator Author

mewall commented Mar 7, 2022 via email

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.

5 participants