Add gpu tag for ROCm-LLVM and refactor method#228
Conversation
| add_gpu_property = '' | ||
|
|
||
| # If none of the gpu packages are in the easyconfig, do not process further | ||
| dep_names = {dep[0] for dep in ec_dict['dependencies']} |
There was a problem hiding this comment.
Have you tested this? I wonder if ROCm-LLVM is the toolchain if it actually appears in the dependencies?
There was a problem hiding this comment.
I doubt it. For one of the other hooks (can't remember which) where we searched some dep list with CUDA / cuDNN in it, I added logic to also check the toolchain instead - and then just trigger it if it's rocm-compilers, rompi,... etc. This should probably do something similar. @zerefwayne search for rocm-compilers in eb_hooks.py and you'll probably find my logic in the other hook.
There was a problem hiding this comment.
It would be better if we can identify ROCm-LLVM explicitly so that we can store the version in an environment variable. That will be useful later in Lmod when we are checking if the version is supported by the driver.
3 participants
Easyconfigs which include ROCm-LLVM as a dependency should also be tagged with
gpu. It doesn't need to be dropped to a build dependency as it is redistributable (unlike CUDA).The loop seems to iterate over dependencies twice, it can be simplified to one pass.