File tree Expand file tree Collapse file tree 1 file changed +9
-0
lines changed
Expand file tree Collapse file tree 1 file changed +9
-0
lines changed Original file line number Diff line number Diff line change @@ -105,6 +105,15 @@ if (eessi_accelerator_target ~= nil) then
105105 else
106106 LmodError("Incorrect value for $EESSI_ACCELERATOR_TARGET: " .. eessi_accelerator_target)
107107 end
108+
109+ -- If architectures are 9.0, 10.0 or 12.0, enable architecture or family-specific optimizations
110+ if easybuild_cuda_compute_capabilities == '9.0' then
111+ easybuild_cuda_compute_capabilities = '9.0a'
112+ elseif easybuild_cuda_compute_capabilities == '10.0' then
113+ easybuild_cuda_compute_capabilities = '10.0f'
114+ elseif easybuild_cuda_compute_capabilities == '12.0' then
115+ easybuild_cuda_compute_capabilities = '12.0f'
116+ end
108117end
109118
110119-- Some environment variables affect behaviour, let's gather them once
You can’t perform that action at this time.
0 commit comments