Tim Dortmann
141 lines
4.3 KiB
Nix
141 lines
4.3 KiB
Nix
{
|
|
backendStdenv,
|
|
buildRedist,
|
|
cudaAtLeast,
|
|
cudaMajorMinorVersion,
|
|
cudaOlder,
|
|
e2fsprogs,
|
|
elfutils,
|
|
flags,
|
|
gst_all_1,
|
|
lib,
|
|
libjpeg8,
|
|
qt5 ? null,
|
|
qt6 ? null,
|
|
rdma-core,
|
|
ucx,
|
|
}:
|
|
let
|
|
archDir =
|
|
{
|
|
aarch64-linux = "linux-" + (if flags.isJetsonBuild then "v4l_l4t" else "desktop") + "-t210-a64";
|
|
x86_64-linux = "linux-desktop-glibc_2_11_3-x64";
|
|
}
|
|
.${backendStdenv.hostPlatform.system}
|
|
or (throw "Unsupported system: ${backendStdenv.hostPlatform.system}");
|
|
in
|
|
buildRedist (
|
|
finalAttrs:
|
|
let
|
|
qt = if lib.versionOlder finalAttrs.version "2022.2.0" then qt5 else qt6;
|
|
qtwayland =
|
|
if lib.versions.major qt.qtbase.version == "5" then
|
|
lib.getBin qt.qtwayland
|
|
else
|
|
lib.getLib qt.qtwayland;
|
|
inherit (qt) wrapQtAppsHook qtwebengine qtwebview;
|
|
in
|
|
{
|
|
redistName = "cuda";
|
|
pname = "nsight_compute";
|
|
|
|
# NOTE(@connorbaker): It might be a problem that when nsight_compute contains hosts and targets of different
|
|
# architectures, that we patchelf just the binaries matching the builder's platform; autoPatchelfHook prints
|
|
# messages like
|
|
# skipping [$out]/host/linux-desktop-glibc_2_11_3-x64/libQt6Core.so.6 because its architecture (x64) differs from
|
|
# target (AArch64)
|
|
outputs = [ "out" ];
|
|
|
|
allowFHSReferences = true;
|
|
|
|
nativeBuildInputs = [ wrapQtAppsHook ];
|
|
|
|
buildInputs = [
|
|
(lib.getLib qtwayland)
|
|
qtwebengine
|
|
qtwebview
|
|
rdma-core
|
|
]
|
|
++ lib.optionals (cudaMajorMinorVersion == "12.0" && backendStdenv.hostPlatform.isAarch64) [
|
|
libjpeg8
|
|
]
|
|
++ lib.optionals (cudaAtLeast "12.1" && cudaOlder "12.4") [
|
|
gst_all_1.gstreamer
|
|
gst_all_1.gst-plugins-base
|
|
]
|
|
++ lib.optionals (cudaAtLeast "12.0" && cudaOlder "12.7") [
|
|
e2fsprogs
|
|
ucx
|
|
]
|
|
++ lib.optionals (cudaAtLeast "12.9") [ elfutils ];
|
|
|
|
dontWrapQtApps = true;
|
|
|
|
preInstall = ''
|
|
if [[ -d nsight-compute ]]; then
|
|
nixLog "Lifting components of Nsight Compute to the top level"
|
|
mv -v nsight-compute/*/* .
|
|
nixLog "Removing empty directories"
|
|
rmdir -pv nsight-compute/*
|
|
fi
|
|
|
|
rm -rf host/${archDir}/Mesa/
|
|
|
|
patchShebangs .
|
|
'';
|
|
|
|
postInstall = ''
|
|
moveToOutput 'ncu' "''${!outputBin}/bin"
|
|
moveToOutput 'ncu-ui' "''${!outputBin}/bin"
|
|
moveToOutput 'host' "''${!outputBin}/bin"
|
|
moveToOutput 'target' "''${!outputBin}/bin"
|
|
moveToOutput 'sections' "''${!outputBin}/bin"
|
|
wrapQtApp "''${!outputBin}/bin/host/${archDir}/ncu-ui.bin"
|
|
''
|
|
# NOTE(@connorbaker): No idea what this platform is or how to patchelf for it.
|
|
+ lib.optionalString (flags.isJetsonBuild && cudaAtLeast "11.8" && cudaOlder "12.9") ''
|
|
nixLog "Removing QNX 700 target directory for Jetson builds"
|
|
rm -rfv "''${!outputBin}/bin/target/qnx-700-t210-a64"
|
|
''
|
|
+ lib.optionalString (flags.isJetsonBuild && cudaAtLeast "12.8") ''
|
|
nixLog "Removing QNX 800 target directory for Jetson builds"
|
|
rm -rfv "''${!outputBin}/bin/target/qnx-800-tegra-a64"
|
|
'';
|
|
|
|
# lib needs libtiff.so.5, but nixpkgs provides libtiff.so.6
|
|
preFixup = ''
|
|
patchelf --replace-needed libtiff.so.5 libtiff.so "''${!outputBin}/bin/host/${archDir}/Plugins/imageformats/libqtiff.so"
|
|
'';
|
|
|
|
autoPatchelfIgnoreMissingDeps = [
|
|
"libcuda.so.1"
|
|
"libnvidia-ml.so.1"
|
|
];
|
|
|
|
brokenAssertions = [
|
|
{
|
|
message = "Qt 5 is required and available";
|
|
assertion = lib.versionOlder finalAttrs.version "2022.2.0" -> qt5 != null;
|
|
}
|
|
{
|
|
message = "Qt 6 is required and available";
|
|
assertion = lib.versionAtLeast finalAttrs.version "2022.2.0" -> qt6 != null;
|
|
}
|
|
];
|
|
|
|
meta = {
|
|
description = "Interactive profiler for CUDA and NVIDIA OptiX";
|
|
longDescription = ''
|
|
NVIDIA Nsight Compute is an interactive profiler for CUDA and NVIDIA OptiX that provides detailed performance
|
|
metrics and API debugging via a user interface and command-line tool. Users can run guided analysis and compare
|
|
results with a customizable and data-driven user interface, as well as post-process and analyze results in their
|
|
own workflows.
|
|
'';
|
|
homepage = "https://developer.nvidia.com/nsight-compute";
|
|
changelog = "https://docs.nvidia.com/nsight-compute/ReleaseNotes";
|
|
|
|
mainProgram = "ncu";
|
|
};
|
|
}
|
|
)
|