pyrox.dev
magma: symlinkJoin -> CUDAToolkit_ROOT
+10
-27
pkgs/development/libraries/science/math/magma/generic.nix
+10
-27
pkgs/development/libraries/science/math/magma/generic.nix
···
86
86
# "75" -> "750" Cf. https://bitbucket.org/icl/magma/src/f4ec79e2c13a2347eff8a77a3be6f83bc2daec20/CMakeLists.txt#lines-273
87
87
"${minArch'}0";
88
88
89
-
cuda-common-redist = with cudaPackages; [
90
-
libcublas # cublas_v2.h
91
-
libcusparse # cusparse.h
92
-
];
93
-
94
-
# Build-time dependencies
95
-
cuda-native-redist = symlinkJoin {
96
-
name = "cuda-native-redist-${cudaVersion}";
97
-
paths = with cudaPackages; [
98
-
cuda_cudart # cuda_runtime.h
99
-
cuda_nvcc
100
-
] ++ lists.optionals (strings.versionOlder cudaVersion "11.8") [
101
-
cuda_nvprof # <cuda_profiler_api.h>
102
-
] ++ lists.optionals (strings.versionAtLeast cudaVersion "11.8") [
103
-
cuda_profiler_api # <cuda_profiler_api.h>
104
-
] ++ cuda-common-redist;
105
-
};
106
-
107
-
# Run-time dependencies
108
-
cuda-redist = symlinkJoin {
109
-
name = "cuda-redist-${cudaVersion}";
110
-
paths = cuda-common-redist;
111
-
};
112
89
in
113
90
114
91
assert (builtins.match "[^[:space:]]*" gpuTargetString) != null;
···
128
105
ninja
129
106
gfortran
130
107
] ++ lists.optionals cudaSupport [
131
-
cuda-native-redist
108
+
cudaPackages.cuda_nvcc
132
109
];
133
110
134
111
buildInputs = [
135
112
libpthreadstubs
136
113
lapack
137
114
blas
138
-
] ++ lists.optionals cudaSupport [
139
-
cuda-redist
140
-
] ++ lists.optionals rocmSupport [
115
+
] ++ lists.optionals cudaSupport (with cudaPackages; [
116
+
cuda_cudart
117
+
libcublas # cublas_v2.h
118
+
libcusparse # cusparse.h
119
+
] ++ lists.optionals (strings.versionOlder cudaVersion "11.8") [
120
+
cuda_nvprof # <cuda_profiler_api.h>
121
+
] ++ lists.optionals (strings.versionAtLeast cudaVersion "11.8") [
122
+
cuda_profiler_api # <cuda_profiler_api.h>
123
+
]) ++ lists.optionals rocmSupport [
141
124
hip
142
125
hipblas
143
126
hipsparse