abonerib/jungle
1
0
Fork 0
forked from rarias/jungle
Code Pull Requests Activity

Add TACUDA through cudaPackages extensions

Browse Source
This commit is contained in:
Aleix Boné 2025-12-16 14:20:05 +01:00
parent 7c833c556e
commit a329893fbe
No known key found for this signature in database
2 changed files with 15 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
overlay.nix
View File

@ -59,11 +59,16 @@ let
stdenvClangOmpss2Nanos6 = final.stdenv.override { cc = final.clangOmpss2Nanos6; allowedRequisites = null; }; stdenvClangOmpss2Nanos6 = final.stdenv.override { cc = final.clangOmpss2Nanos6; allowedRequisites = null; };
stdenvClangOmpss2Nodes = final.stdenv.override { cc = final.clangOmpss2Nodes; allowedRequisites = null; }; stdenvClangOmpss2Nodes = final.stdenv.override { cc = final.clangOmpss2Nodes; allowedRequisites = null; };
stdenvClangOmpss2NodesOmpv = final.stdenv.override { cc = final.clangOmpss2NodesOmpv; allowedRequisites = null; }; stdenvClangOmpss2NodesOmpv = final.stdenv.override { cc = final.clangOmpss2NodesOmpv; allowedRequisites = null; };
tacuda = callPackage ./pkgs/tacuda/default.nix { };
tagaspi = callPackage ./pkgs/tagaspi/default.nix { }; tagaspi = callPackage ./pkgs/tagaspi/default.nix { };
tampi = callPackage ./pkgs/tampi/default.nix { }; tampi = callPackage ./pkgs/tampi/default.nix { };
upc-qaire-exporter = prev.callPackage ./pkgs/upc-qaire-exporter/default.nix { }; upc-qaire-exporter = prev.callPackage ./pkgs/upc-qaire-exporter/default.nix { };
wxparaver = callPackage ./pkgs/paraver/default.nix { }; wxparaver = callPackage ./pkgs/paraver/default.nix { };
_cuda = prev._cuda.extend (_: _prev: final.lib.recursiveUpdate _prev {
extensions = _prev.extensions ++ [(finalAttrs: _: {
tacuda = finalAttrs.callPackage ./pkgs/tacuda/default.nix { };
})];
});
}; };
tests = rec { tests = rec {

15
pkgs/tacuda/default.nix
View File

@ -1,5 +1,5 @@
{ {
stdenv, backendStdenv,
fetchFromGitHub, fetchFromGitHub,
automake, automake,
autoconf, autoconf,
@ -7,7 +7,9 @@
gnumake, gnumake,
autoreconfHook, autoreconfHook,
boost, boost,
cudaPackages, cudatoolkit,
libcublas,
cuda_cudart,
useGit ? false, useGit ? false,
gitUrl ? "git@gitlab-internal.bsc.es:task-awareness/tacuda/tacuda.git", gitUrl ? "git@gitlab-internal.bsc.es:task-awareness/tacuda/tacuda.git",
@ -41,12 +43,13 @@ let
in in
stdenv.mkDerivation { backendStdenv.mkDerivation {
pname = "tacuda"; pname = "tacuda";
inherit (source) src version; inherit (source) src version;
enableParallelBuilding = true; enableParallelBuilding = true;
separateDebugInfo = true; separateDebugInfo = true;
strictDeps = true;
nativeBuildInputs = [ nativeBuildInputs = [
autoreconfHook autoreconfHook
@ -58,12 +61,12 @@ stdenv.mkDerivation {
patches = [ ./fix_config.patch ]; patches = [ ./fix_config.patch ];
configureFlags = [ "--with-cuda-include=${cudaPackages.cudatoolkit}/include" ]; configureFlags = [ "--with-cuda-include=${cudatoolkit}/include" ];
buildInputs = [ buildInputs = [
boost boost
cudaPackages.libcublas libcublas
cudaPackages.cuda_cudart cuda_cudart
]; ];
} }