zfp: nixfmt

This commit is contained in:
wxt
2024-10-30 22:35:29 +08:00
parent a6cba048e6
commit 412f8116f9
+31 -12
View File
@@ -1,11 +1,20 @@
{ cmake, cudatoolkit, fetchFromGitHub, gfortran, lib, llvmPackages, python3Packages, stdenv
, config
, enableCfp ? true
, enableCuda ? config.cudaSupport
, enableFortran ? builtins.elem stdenv.hostPlatform.system gfortran.meta.platforms
, enableOpenMP ? true
, enablePython ? true
, enableUtilities ? true }:
{
cmake,
cudatoolkit,
fetchFromGitHub,
gfortran,
lib,
llvmPackages,
python3Packages,
stdenv,
config,
enableCfp ? true,
enableCuda ? config.cudaSupport,
enableFortran ? builtins.elem stdenv.hostPlatform.system gfortran.meta.platforms,
enableOpenMP ? true,
enablePython ? true,
enableUtilities ? true,
}:
stdenv.mkDerivation rec {
pname = "zfp";
@@ -20,10 +29,18 @@ stdenv.mkDerivation rec {
nativeBuildInputs = [ cmake ];
buildInputs = lib.optional enableCuda cudatoolkit
buildInputs =
lib.optional enableCuda cudatoolkit
++ lib.optional enableFortran gfortran
++ lib.optional enableOpenMP llvmPackages.openmp
++ lib.optionals enablePython (with python3Packages; [ cython numpy python ]);
++ lib.optionals enablePython (
with python3Packages;
[
cython
numpy
python
]
);
# compile CUDA code for all extant GPUs so the binary will work with any GPU
# and driver combination. to be ultimately solved upstream:
@@ -35,8 +52,10 @@ stdenv.mkDerivation rec {
)
'';
cmakeFlags = [
] ++ lib.optional enableCfp "-DBUILD_CFP=ON"
cmakeFlags =
[
]
++ lib.optional enableCfp "-DBUILD_CFP=ON"
++ lib.optional enableCuda "-DZFP_WITH_CUDA=ON"
++ lib.optional enableFortran "-DBUILD_ZFORP=ON"
++ lib.optional enableOpenMP "-DZFP_WITH_OPENMP=ON"