feat: Added nix build support
Some checks failed
build / checks-matrix (push) Successful in 19m19s
build / codecov (push) Failing after 19m26s
docs / docs (push) Has been cancelled
build / checks-build (push) Has been cancelled

This commit is contained in:
uttarayan21
2025-08-19 14:30:43 +05:30
parent cd12e97de3
commit 3d56db687c
3 changed files with 23 additions and 22 deletions

View File

@@ -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 = {