Newer
Older
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
packages:
all:
require:
- "~rocm"
- spec: cuda_arch={cuda_arch}
when: '+cuda'
- spec: '+cuda'
when: '%gcc'
# ---------------------------------------------------------------------------
cuda:
prefer:
- spec: "^gcc-runtime@{gcc_version}"
when: "%gcc"
- spec: "%gcc@{gcc_version} target={target}"
when: "%gcc"
require:
- '@12.4.1'
hypre:
require:
- spec: '+cuda cuda_arch={cuda_arch} +unified-memory'
when: '%gcc'
kokkos:
require:
- spec: '+cuda +cuda_lambda cuda_arch={cuda_arch} +wrapper ~openmptarget'
when: '%gcc'
kokkos-kernels:
require: ['+cuda', '+cublas', '+cusolver', '+cusparse', 'cuda_arch={cuda_arch}']
libxc:
require: ['+cuda', 'cuda_arch={cuda_arch}']
nvshmem:
require:
- spec: '+cuda cuda_arch={cuda_arch} +ibgda +nccl +ucx +mpi'
openmpi:
require: ['+cuda', 'cuda_arch={cuda_arch}', 'fabrics=hcoll']
papi:
require: ['+nvml', '+infiniband', '+cuda']
paraview:
require: ['~cuda']
petsc:
require:
- spec: '+cuda cuda_arch={cuda_arch}'
when: '%gcc'
py-tensorflow:
require:
- spec: '+cuda cuda_arch={cuda_arch} +nccl'
py-torch:
require:
- spec: '+cuda cuda_arch={cuda_arch} +nccl +cudnn'
python:
require:
- spec: 'target={target}'
when: '%gcc@{gcc_version}'
quantum-espresso:
require:
- spec: '+nvtx'
suite-sparse:
require:
- spec: '+cuda'
when: '@7.3.1'
- spec: '~cuda'
when: '@7.2.1'
ucc:
require:
- spec: '+cuda cuda_arch={cuda_arch} +nccl'
ucx:
require:
- spec: '+xpmem +cma +rdmacm +rc +ud +dc +verbs +ib_hw_tm +gdrcopy +cuda cuda_arch={cuda_arch} ~rocm'