onnxruntime: only set NIX_LDFLAGS on linux (#482865)

This commit is contained in:
Yureka
2026-01-23 13:51:30 +00:00
committed by GitHub
2 changed files with 24 additions and 23 deletions
+23 -22
View File
@@ -317,28 +317,29 @@ effectiveStdenv.mkDerivation (finalAttrs: {
(lib.cmakeBool "onnxruntime_USE_COMPOSABLE_KERNEL_CK_TILE" false)
];
env = {
NIX_LDFLAGS = "-z,noexecstack";
}
// lib.optionalAttrs rocmSupport {
MIOPEN_PATH = rocmPackages.miopen;
# HIP steps fail to find ROCm libs when not in HIPFLAGS, causing
# fatal error: 'rocrand/rocrand.h' file not found
HIPFLAGS = lib.concatMapStringsSep " " (pkg: "-I${lib.getInclude pkg}/include") [
rocmPackages.hipblas
rocmPackages.hipcub
rocmPackages.hiprand
rocmPackages.hipsparse
rocmPackages.rocblas
rocmPackages.rocprim
rocmPackages.rocrand
rocmPackages.rocthrust
];
}
// lib.optionalAttrs effectiveStdenv.hostPlatform.isMusl {
NIX_CFLAGS_COMPILE = "-DFLATBUFFERS_LOCALE_INDEPENDENT=0";
GTEST_FILTER = "*:-ContribOpTest.StringNormalizer*";
};
env =
lib.optionalAttrs effectiveStdenv.hostPlatform.isLinux {
NIX_LDFLAGS = "-z,noexecstack";
}
// lib.optionalAttrs rocmSupport {
MIOPEN_PATH = rocmPackages.miopen;
# HIP steps fail to find ROCm libs when not in HIPFLAGS, causing
# fatal error: 'rocrand/rocrand.h' file not found
HIPFLAGS = lib.concatMapStringsSep " " (pkg: "-I${lib.getInclude pkg}/include") [
rocmPackages.hipblas
rocmPackages.hipcub
rocmPackages.hiprand
rocmPackages.hipsparse
rocmPackages.rocblas
rocmPackages.rocprim
rocmPackages.rocrand
rocmPackages.rocthrust
];
}
// lib.optionalAttrs effectiveStdenv.hostPlatform.isMusl {
NIX_CFLAGS_COMPILE = "-DFLATBUFFERS_LOCALE_INDEPENDENT=0";
GTEST_FILTER = "*:-ContribOpTest.StringNormalizer*";
};
doCheck =
!(
@@ -38,7 +38,7 @@ buildPythonPackage {
chmod +w dist
'';
env = {
env = lib.optionalAttrs stdenv.hostPlatform.isLinux {
NIX_LDFLAGS = "-z,noexecstack";
};