-
Notifications
You must be signed in to change notification settings - Fork 21
Fixes to compile/run on aurora #264
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
base: main
Are you sure you want to change the base?
Conversation
|
Documentation preview: https://applied-material-modeling.github.io/neml2/pr-preview/pr-264 |
| set(NEML2_DOC OFF CACHE BOOL "Build NEML2 documentation (html)") | ||
| set(NEML2_WORK_DISPATCHER OFF CACHE BOOL "Enable NEML2 work dispatcher") | ||
| set(NEML2_JSON OFF CACHE BOOL "Enable JSON support") | ||
| set(AURORA_BUILD OFF CACHE BOOL "Build on ALCF Aurora machine") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'd prefer a more general flag like NEML2_IPEX. I suppose there's nothing else specific to Aurora.
| # Intel Extension for PyTorch | ||
| # ---------------------------------------------------------------------------- | ||
| if (AURORA_BUILD) | ||
| find_library(IPEX_LIB intel-ext-pt-gpu PATHS ${INTEL_EXTENSION_FOR_PYTORCH_PATH}/lib NO_DEFAULT_PATH REQUIRED) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
| find_library(IPEX_LIB intel-ext-pt-gpu PATHS ${INTEL_EXTENSION_FOR_PYTORCH_PATH}/lib NO_DEFAULT_PATH REQUIRED) | |
| find_library(IPEX_LIB intel-ext-pt-gpu) |
I suspect something like this will just work. You can then specify the path using -DIPEX_LIB_ROOT=....
| if (AURORA_BUILD) | ||
| find_package(Threads REQUIRED) | ||
| target_link_libraries(dispatcher_tests PRIVATE testutils Threads::Threads) | ||
| else() | ||
| target_link_libraries(dispatcher_tests PRIVATE testutils) | ||
| endif() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why is this required?
Miscellaneous fixes required to get things to work on aurora.