Conversation
| value = 'add_property("arch","gpu")' | ||
| cuda_version = 0 | ||
| for dep in iter(ec_dict["dependencies"]): | ||
| # Make CUDA a build dependency only (rpathing saves us from link errors) |
There was a problem hiding this comment.
This approach saves us from explicitly loading CUDA to run a CUDA dependent package. This allows us to write an Lmod hook that protects loading the CUDA module unless certain criteria are met (i.e., that the symlinks are unbroken).
| ): | ||
| ec.log.info("[parse hook] Injecting gpu as Lmod arch property and envvar with CUDA version") | ||
| key = "modluafooter" | ||
| value = 'add_property("arch","gpu")' |
There was a problem hiding this comment.
This property allows us to protect the loading of any GPU package via an Lmod hook (which can be overridden): unless the compat libraries are installed you can't load GPU modules
eb_hooks.py
Outdated
There was a problem hiding this comment.
We should be checking that these are succeeding
|
This is working but requires an additional script to unbreak the symlinks in the CUDA installation (which is being extracted from #172 ) |
|
This requires easybuilders/easybuild-framework#4119 or you install the compat libraries so that the rpath check passes |
install_cuda_host_injections.sh
Outdated
There was a problem hiding this comment.
Should allow this to be overridden
install_cuda_host_injections.sh
Outdated
There was a problem hiding this comment.
The parent dir also needs to be writable.
This will allow us to log where creating directory structures under `host_injections` is breaking down.
install_cuda_host_injections.sh
Outdated
There was a problem hiding this comment.
This assumes that ${cuda_install_dir} exists, which it may not
There was a problem hiding this comment.
In general this space check is a bit clumsy, it's assuming everything happens in one location but there are actually 3 (source, build, install) each of which takes ~5GB
|
@ocaisa Conflicts to fix. Shall we re-target this to the (new) |
…ld/4.8.2
{2023.06}[system] EasyBuild V4.8.2
|
GPU support implemented with #434 |
2 participants
This is part of a logical splitting of #172 to make it a bit more manageable
Requires #228