From 48507397d0d7277407ddd39e914c583f69bdf048 Mon Sep 17 00:00:00 2001
From: Someone Serge <sergei.kozlukov@aalto.fi>
Date: Wed, 28 Aug 2024 11:36:32 +0000
Subject: [PATCH] nix: fix: mpi-cuda should include cuda support...

---
 flake.nix | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/flake.nix b/flake.nix
index c69637d11..a5617d37a 100644
--- a/flake.nix
+++ b/flake.nix
@@ -162,7 +162,7 @@
                 cuda = config.legacyPackages.llamaPackagesCuda.llama-cpp;
 
                 mpi-cpu = config.packages.default.override { useMpi = true; };
-                mpi-cuda = config.packages.default.override { useMpi = true; };
+                mpi-cuda = config.packages.cuda.override { useMpi = true; };
               }
               // lib.optionalAttrs (system == "x86_64-linux") {
                 rocm = config.legacyPackages.llamaPackagesRocm.llama-cpp;