Skip to content

Commit aae5169

Browse files
author
Caspar van Leeuwen
committed
Enable by default the architecture-specific optimizations for 9.0, and the family specific optimizations for 10.0 and 12.0
1 parent 8dde4c2 commit aae5169

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

EESSI-extend-easybuild.eb

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff 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
108117
end
109118
110119
-- Some environment variables affect behaviour, let's gather them once

0 commit comments

Comments
 (0)