about summary refs log tree commit diff
path: root/nixpkgs/pkgs/by-name/si/sirius/package.nix
diff options
context:
space:
mode:
Diffstat (limited to 'nixpkgs/pkgs/by-name/si/sirius/package.nix')
-rw-r--r--nixpkgs/pkgs/by-name/si/sirius/package.nix9
1 files changed, 8 insertions, 1 deletions
diff --git a/nixpkgs/pkgs/by-name/si/sirius/package.nix b/nixpkgs/pkgs/by-name/si/sirius/package.nix
index 2af3c28de922..8518092203b8 100644
--- a/nixpkgs/pkgs/by-name/si/sirius/package.nix
+++ b/nixpkgs/pkgs/by-name/si/sirius/package.nix
@@ -21,9 +21,16 @@
 , eigen
 , libvdwxc
 , llvmPackages
-, gpuBackend ? "none"
 , cudaPackages
 , rocmPackages
+, config
+, gpuBackend ? (
+  if config.cudaSupport
+  then "cuda"
+  else if config.rocmSupport
+  then "rocm"
+  else "none"
+)
 }:
 
 assert builtins.elem gpuBackend [ "none" "cuda" "rocm" ];