feat: Added nix build support
This commit is contained in:
26
flake.nix
26
flake.nix
@@ -78,6 +78,14 @@
|
||||
craneLib = (crane.mkLib pkgs).overrideToolchain stableToolchain;
|
||||
craneLibLLvmTools = (crane.mkLib pkgs).overrideToolchain stableToolchainWithLLvmTools;
|
||||
|
||||
ort_static = pkgs.onnxruntime.overrideAttrs (old: {
|
||||
cmakeFlags =
|
||||
old.cmakeFlags
|
||||
++ [
|
||||
"-Donnxruntime_BUILD_SHARED_LIB=OFF"
|
||||
"-Donnxruntime_BUILD_STATIC_LIB=ON"
|
||||
];
|
||||
});
|
||||
src = let
|
||||
filterBySuffix = path: exts: lib.any (ext: lib.hasSuffix ext path) exts;
|
||||
sourceFilters = path: type:
|
||||
@@ -96,25 +104,16 @@
|
||||
filter = sourceFilters;
|
||||
src = ./.;
|
||||
};
|
||||
commonArgs = let
|
||||
ort_static = pkgs.onnxruntime.overrideAttrs (old: {
|
||||
cmakeFlags =
|
||||
old.cmakeFlags
|
||||
++ [
|
||||
"-Donnxruntime_BUILD_SHARED_LIB=OFF"
|
||||
"-Donnxruntime_BUILD_STATIC_LIB=ON"
|
||||
];
|
||||
});
|
||||
in
|
||||
commonArgs =
|
||||
{
|
||||
inherit src;
|
||||
pname = name;
|
||||
stdenv = p: p.clangStdenv;
|
||||
doCheck = false;
|
||||
LIBCLANG_PATH = "${pkgs.llvmPackages.libclang.lib}/lib";
|
||||
ORT_LIB_LOCATION = "${ort_static}";
|
||||
ORT_ENV_SYSTEM_LIB_LOCATION = "${ort_static}/lib";
|
||||
# ORT_ENV_PREFER_DYNAMIC_LINK = true;
|
||||
ORT_LIB_LOCATION = "${pkgs.onnxruntime}";
|
||||
ORT_ENV_SYSTEM_LIB_LOCATION = "${pkgs.onnxruntime}/lib";
|
||||
ORT_ENV_PREFER_DYNAMIC_LINK = true;
|
||||
nativeBuildInputs = with pkgs; [
|
||||
cmake
|
||||
pkg-config
|
||||
@@ -192,6 +191,7 @@
|
||||
in {
|
||||
"${name}" = pkg;
|
||||
default = pkg;
|
||||
onnxruntime = ort_static;
|
||||
};
|
||||
|
||||
devShells = {
|
||||
|
||||
Reference in New Issue
Block a user