646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
137 lines
3.7 KiB
Nix
137 lines
3.7 KiB
Nix
{
|
|
boost178,
|
|
cuda_cudart,
|
|
cudaAtLeast,
|
|
e2fsprogs,
|
|
gst_all_1,
|
|
lib,
|
|
nss,
|
|
numactl,
|
|
pulseaudio,
|
|
qt6,
|
|
rdma-core,
|
|
stdenv,
|
|
ucx,
|
|
wayland,
|
|
xorg,
|
|
}:
|
|
prevAttrs:
|
|
let
|
|
qtwayland = lib.getLib qt6.qtwayland;
|
|
qtWaylandPlugins = "${qtwayland}/${qt6.qtbase.qtPluginPrefix}";
|
|
# NOTE(@connorbaker): nsight_systems doesn't support Jetson, so no need for case splitting on aarch64-linux.
|
|
hostDir =
|
|
{
|
|
aarch64-linux = "host-linux-armv8";
|
|
x86_64-linux = "host-linux-x64";
|
|
}
|
|
.${stdenv.hostPlatform.system} or (throw "Unsupported system: ${stdenv.hostPlatform.system}");
|
|
targetDir =
|
|
{
|
|
aarch64-linux = "target-linux-sbsa-armv8";
|
|
x86_64-linux = "target-linux-x64";
|
|
}
|
|
.${stdenv.hostPlatform.system} or (throw "Unsupported system: ${stdenv.hostPlatform.system}");
|
|
in
|
|
{
|
|
outputs = [ "out" ]; # NOTE(@connorbaker): Force a single output so relative lookups work.
|
|
|
|
# An ad hoc replacement for
|
|
# https://github.com/ConnorBaker/cuda-redist-find-features/issues/11
|
|
env = prevAttrs.env or { } // {
|
|
rmPatterns =
|
|
prevAttrs.env.rmPatterns or ""
|
|
+ toString [
|
|
"${hostDir}/lib{arrow,jpeg}*"
|
|
"${hostDir}/lib{ssl,ssh,crypto}*"
|
|
"${hostDir}/libboost*"
|
|
"${hostDir}/libexec"
|
|
"${hostDir}/libstdc*"
|
|
"${hostDir}/python/bin/python"
|
|
"${hostDir}/Mesa"
|
|
];
|
|
};
|
|
|
|
# NOTE(@connorbaker): nsight-exporter and nsight-sys are deprecated scripts wrapping nsys, it's fine to remove them.
|
|
prePatch = prevAttrs.prePatch or "" + ''
|
|
if [[ -d bin ]]; then
|
|
nixLog "Removing bin wrapper scripts"
|
|
for knownWrapper in bin/{nsys{,-ui},nsight-{exporter,sys}}; do
|
|
[[ -e $knownWrapper ]] && rm -v "$knownWrapper"
|
|
done
|
|
unset -v knownWrapper
|
|
|
|
nixLog "Removing empty bin directory"
|
|
rmdir -v bin
|
|
fi
|
|
|
|
if [[ -d nsight-systems ]]; then
|
|
nixLog "Lifting components of Nsight System to the top level"
|
|
mv -v nsight-systems/*/* .
|
|
nixLog "Removing empty nsight-systems directory"
|
|
rmdir -pv nsight-systems/*
|
|
fi
|
|
'';
|
|
|
|
postPatch = prevAttrs.postPatch or "" + ''
|
|
for path in $rmPatterns; do
|
|
rm -r "$path"
|
|
done
|
|
patchShebangs nsight-systems
|
|
'';
|
|
|
|
nativeBuildInputs = prevAttrs.nativeBuildInputs or [ ] ++ [ qt6.wrapQtAppsHook ];
|
|
|
|
dontWrapQtApps = true;
|
|
|
|
buildInputs =
|
|
prevAttrs.buildInputs or [ ]
|
|
++ [
|
|
(qt6.qtdeclarative or qt6.full)
|
|
(qt6.qtsvg or qt6.full)
|
|
(qt6.qtimageformats or qt6.full)
|
|
(qt6.qtpositioning or qt6.full)
|
|
(qt6.qtscxml or qt6.full)
|
|
(qt6.qttools or qt6.full)
|
|
(qt6.qtwebengine or qt6.full)
|
|
(qt6.qtwayland or qt6.full)
|
|
boost178
|
|
cuda_cudart.stubs
|
|
e2fsprogs
|
|
gst_all_1.gst-plugins-base
|
|
gst_all_1.gstreamer
|
|
nss
|
|
numactl
|
|
pulseaudio
|
|
qt6.qtbase
|
|
qtWaylandPlugins
|
|
rdma-core
|
|
ucx
|
|
wayland
|
|
xorg.libXcursor
|
|
xorg.libXdamage
|
|
xorg.libXrandr
|
|
xorg.libXtst
|
|
]
|
|
# NOTE(@connorbaker): Seems to be required only for aarch64-linux.
|
|
++ lib.optionals stdenv.hostPlatform.isAarch64 [
|
|
gst_all_1.gst-plugins-bad
|
|
];
|
|
|
|
postInstall = prevAttrs.postInstall or "" + ''
|
|
moveToOutput '${hostDir}' "''${!outputBin}"
|
|
moveToOutput '${targetDir}' "''${!outputBin}"
|
|
moveToOutput 'bin' "''${!outputBin}"
|
|
wrapQtApp "''${!outputBin}/${hostDir}/nsys-ui.bin"
|
|
'';
|
|
|
|
# lib needs libtiff.so.5, but nixpkgs provides libtiff.so.6
|
|
preFixup = prevAttrs.preFixup or "" + ''
|
|
patchelf --replace-needed libtiff.so.5 libtiff.so "''${!outputBin}/${hostDir}/Plugins/imageformats/libqtiff.so"
|
|
'';
|
|
|
|
autoPatchelfIgnoreMissingDeps = prevAttrs.autoPatchelfIgnoreMissingDeps or [ ] ++ [
|
|
"libnvidia-ml.so.1"
|
|
];
|
|
}
|