forked from rarias/jungle
Add unified-runtime with CUDA
This commit is contained in:
parent
cb5b730458
commit
0fabb9da19
@ -79,6 +79,7 @@ let
|
||||
tampi = callPackage ./pkgs/tampi/default.nix { };
|
||||
tasycl = callPackage ./pkgs/tasycl/default.nix { };
|
||||
tasycl-acpp = callPackage ./pkgs/tasycl/default.nix { useIntel = false; };
|
||||
unified-runtime = callPackage ./pkgs/unified-runtime/default.nix { };
|
||||
upc-qaire-exporter = prev.callPackage ./pkgs/upc-qaire-exporter/default.nix { };
|
||||
wxparaver = callPackage ./pkgs/paraver/default.nix { };
|
||||
};
|
||||
|
||||
63
pkgs/unified-runtime/default.nix
Normal file
63
pkgs/unified-runtime/default.nix
Normal file
@ -0,0 +1,63 @@
|
||||
{
|
||||
lib,
|
||||
stdenv,
|
||||
fetchFromGitHub,
|
||||
cmake,
|
||||
python3,
|
||||
cudatoolkit,
|
||||
libbacktrace,
|
||||
intel-compute-runtime,
|
||||
level-zero,
|
||||
unified-memory-framework,
|
||||
autoAddDriverRunpath,
|
||||
|
||||
enableL0 ? false,
|
||||
}:
|
||||
|
||||
stdenv.mkDerivation rec {
|
||||
pname = "unified-runtime";
|
||||
version = "2025-WW45";
|
||||
|
||||
src = fetchFromGitHub {
|
||||
owner = "intel";
|
||||
repo = "llvm";
|
||||
tag = version;
|
||||
sha256 = "sha256-yf8pFEeNEWf9LpEhAy4vMdFDlA08x5XaCegnq+5aXRw=";
|
||||
};
|
||||
|
||||
sourceRoot = "${src.name}/unified-runtime";
|
||||
|
||||
nativeBuildInputs = [
|
||||
cmake
|
||||
python3
|
||||
libbacktrace
|
||||
(python3.withPackages (
|
||||
python-pkgs: with python-pkgs; [
|
||||
lit
|
||||
filecheck
|
||||
]
|
||||
))
|
||||
autoAddDriverRunpath
|
||||
];
|
||||
|
||||
buildInputs = [
|
||||
unified-memory-framework
|
||||
cudatoolkit
|
||||
]
|
||||
++ lib.optionals enableL0 [
|
||||
intel-compute-runtime
|
||||
level-zero
|
||||
];
|
||||
|
||||
cmakeFlags = [
|
||||
(lib.cmakeBool "UR_USE_EXTERNAL_UMF" true)
|
||||
(lib.cmakeBool "UR_BUILD_TESTS" false)
|
||||
|
||||
(lib.cmakeBool "UR_BUILD_ADAPTER_CUDA" true)
|
||||
(lib.cmakeBool "UR_BUILD_ADAPTER_L0" enableL0)
|
||||
|
||||
(lib.cmakeBool "UR_BUILD_ADAPTER_OPENCL" false)
|
||||
(lib.cmakeBool "UR_BUILD_ADAPTER_HIP" false)
|
||||
(lib.cmakeBool "UR_BUILD_ADAPTER_NATIVE_CPU" false)
|
||||
];
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user