Newer
Older
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} +nccl +ucx +mpi'
- spec: '+ibgda'
when: '@3:'
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
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}'
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: ['+gdrcopy', '+cuda', 'cuda_arch={cuda_arch}', '~rocm']