Redo default.nix

Based on: https://github.com/NixOS/nixpkgs/blob/bfb7a882678e518398ce9a31a881538679f6f092/pkgs/development/cuda-modules/saxpy/default.nix
sjmonson · May 24, 2024 · bc89430 · bc89430
1 parent f679336
commit bc89430
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 6 deletions.
diff --git a/default.nix b/default.nix
@@ -1,21 +1,37 @@
 {
   config,
-  stdenv,
   lib,
-  cudaPackages ? { },
+  autoAddDriverRunpath,
+  cudaPackages,
   ...
 }:
-
-stdenv.mkDerivation {
+let
+  inherit (cudaPackages)
+    backendStdenv
+    cuda_cudart
+    cuda_nvcc
+    cudaAtLeast
+    cudaOlder
+    cudatoolkit
+    setupCudaHook
+    ;
+  inherit (lib) getDev getLib getOutput;
+in
+backendStdenv.mkDerivation {
   pname = "tdr-inverse";
   version = "0.0.1";
 
   src = ./.;
 
+  strictDeps = true;
+
   nativeBuildInputs = [
-    cudaPackages.cudatoolkit
+    autoAddDriverRunpath
+    cuda_nvcc
   ];
 
+  buildInputs = [ cuda_cudart ];
+
   # buildPhase = ''
   #   make
   # '';

diff --git a/flake.nix b/flake.nix
@@ -21,7 +21,6 @@
       in
       {
         packages.default = pkgs.callPackage ./. {
-          stdenv = pkgs.gcc11Stdenv;
           cudaPackages = pkgs.cudaPackages_12_3;
         };
-Original file line number
+Diff line change
@@ Expand Up / @@ -21,7 +21,6 @@ @@
           in
           {
             packages.default = pkgs.callPackage ./. {
-              stdenv = pkgs.gcc11Stdenv;
               cudaPackages = pkgs.cudaPackages_12_3;
             };
@@ Expand Down @@