diff options
author | Madoura <madouura@gmail.com> | 2023-10-05 16:58:02 -0500 |
---|---|---|
committer | Madoura <madouura@gmail.com> | 2023-10-09 17:03:19 -0500 |
commit | e6f88a9a824ad8a34bf9d501aa45b0883a4a80c9 (patch) | |
tree | 8a37c8744b64f239268d840a8c143e918c4dd470 /pkgs/by-name/si | |
parent | 6f39d63688276f1953fc1ca3afa16d8bbb7a301d (diff) |
rocm-related: fixup for ROCm 5.7.0
blender: fixup for ROCm 5.7.0 opensycl: fixup for ROCm 5.7.0 magma: fixup for ROCm 5.7.0 torch: fixup for ROCm 5.7.0 cp2k: fixup for ROCm 5.7.0 sirius: fixup for ROCm 5.7.0 spfft: fixup for ROCm 5.7.0 spla: fixup for ROCm 5.7.0
Diffstat (limited to 'pkgs/by-name/si')
-rw-r--r-- | pkgs/by-name/si/sirius/package.nix | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/pkgs/by-name/si/sirius/package.nix b/pkgs/by-name/si/sirius/package.nix index 05d049a7d45b8..2af3c28de9221 100644 --- a/pkgs/by-name/si/sirius/package.nix +++ b/pkgs/by-name/si/sirius/package.nix @@ -23,8 +23,7 @@ , llvmPackages , gpuBackend ? "none" , cudaPackages -, hip -, rocblas +, rocmPackages }: assert builtins.elem gpuBackend [ "none" "cuda" "rocm" ]; @@ -67,8 +66,10 @@ stdenv.mkDerivation rec { libvdwxc ] ++ lib.optional (gpuBackend == "cuda") cudaPackages.cudatoolkit - ++ lib.optionals (gpuBackend == "rocm") [ hip rocblas ] - ++ lib.optional stdenv.isDarwin llvmPackages.openmp + ++ lib.optionals (gpuBackend == "rocm") [ + rocmPackages.clr + rocmPackages.rocblas + ] ++ lib.optional stdenv.isDarwin llvmPackages.openmp ; propagatedBuildInputs = [ mpi ]; @@ -87,7 +88,7 @@ stdenv.mkDerivation rec { ] ++ lib.optionals (gpuBackend == "rocm") [ "-DUSE_ROCM=ON" - "-DHIP_ROOT_DIR=${hip}" + "-DHIP_ROOT_DIR=${rocmPackages.clr}" ]; doCheck = true; |