-
Notifications
You must be signed in to change notification settings - Fork 9
Expand file tree
/
Copy path2020.lua
More file actions
33 lines (30 loc) · 1.09 KB
/
2020.lua
File metadata and controls
33 lines (30 loc) · 1.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
require("SitePackage")
local cc_cluster = os.getenv("CC_CLUSTER") or "computecanada"
local arch = os.getenv("RSNT_ARCH") or ""
local interconnect = os.getenv("RSNT_INTERCONNECT") or ""
local cpu_vendor_id = os.getenv("RSNT_CPU_VENDOR_ID") or ""
if not arch or arch == "" then
if cc_cluster == "cedar" or cc_cluster == "graham" then
arch = "avx2"
else
arch = get_highest_supported_architecture()
end
end
if not cpu_vendor_id or cpu_vendor_id == "" then
cpu_vendor_id = get_cpu_vendor_id()
end
if not interconnect or interconnect == "" then
if cc_cluster == "cedar" then
interconnect = "omnipath"
else
interconnect = get_interconnect()
end
end
local cuda_driver_version = os.getenv("RSNT_CUDA_DRIVER_VERSION") or ""
if not cuda_driver_version or cuda_driver_version == "" then
cuda_driver_version = get_installed_cuda_driver_version()
end
local custom_root = os.getenv("RSNT_CUSTOM_ROOT") or "/cvmfs/soft.computecanada.ca/custom"
assert(loadfile(custom_root .. "/modules/gentoo/2020.lua.core"))(arch, cpu_vendor_id, interconnect, cuda_driver_version)
conflict("StdEnv/2016.4")
conflict("StdEnv/2018.3")