cudaPackages_jetson: rebase onto FindCUDAToolkit.cmake

[SomeoneSerge]

Description of changes

Things done

• Built on platform(s)
  • x86_64-linux
  • aarch64-linux
  • x86_64-darwin
  • aarch64-darwin
• For non-Linux: Is sandbox = true set in nix.conf? (See Nix manual)
• Tested, as applicable:
  • NixOS test(s) (look inside nixos/tests)
  • and/or package tests
  • or, for functions and "core" functionality, tests in lib/tests or pkgs/test
  • made sure NixOS tests are linked to the relevant packages
• Tested compilation of all packages that depend on this change using nix-shell -p nixpkgs-review --run "nixpkgs-review rev HEAD". Note: all changes have to be committed, also see nixpkgs-review usage
• Tested basic functionality of all binary files (usually in ./result/bin/)
• 23.11 Release Notes (or backporting 23.05 Release notes)
  • (Package updates) Added a release notes entry if the change is major or breaking
  • (Module updates) Added a release notes entry if the change is significant
  • (Module addition) Added a release notes entry if adding a new NixOS module
• Fits CONTRIBUTING.md.

[SomeoneSerge]

95b270ed0f21359e5632966b245b22875c64dca3:

❯ nix build -f cross-jetson.nix -L cudaPackages.saxpy
❯ sudo nix store sign --key-file /... $(readlink -f result) --recursive
❯ nix copy --to ssh-ng://jetson-sergei.cs.aalto.fi ./result
someone@ubuntu:~$ LD_LIBRARY_PATH=/usr/lib/aarch64-linux-gnu/tegra /nix/store/sqz64qsr5pxi8y69m7a0xh0imbrhdzi4-saxpy-aarch64-unknown-linux-gnu-unstable-2023-07-11/bin/saxpy
Runtime version: 11040
Driver version: 11040
Max error: 0.000000
❯ nix run -f with-cuda.nix -L cudaPackages.saxpy
Runtime version: 11080
Driver version: 12020
Max error: 0.000000
❯ cat cross-jetson.nix
(import ./. {
  config.allowUnfree = true;
  config.cudaSupport = true;
  config.cudaCapabilities = [ "7.2" ];
  # config.cudaCapabilities = [ "8.6" ];
  overlays = [
    (final: prev: {
      cudaPackages = prev.cudaPackages_11_4.overrideScope' (cu-final: cu-prev: {
        cudaFlags = prev.pkgs.cudaPackages_jetson.cudaFlags;
        cuda_cudart = prev.cudaPackages_11_4.cuda_cudart.override { manifestAttribute = "linux-sbsa"; };
        manifestAttribute = if prev.pkgs.hostPlatform.system == "aarch64-linux" then "linux-aarch64" else null;
      });
      faiss = prev.faiss.override {
        useThrustSourceBuild = false;
      };
      pythonPackagesExtensions = [
        (py-final: py-prev: {
          torch = py-prev.torch.overridePythonAttrs (oldAttrs: { meta = oldAttrs.meta // { broken = false; }; });
        })
      ];
    })
  ];
}).pkgsCross.aarch64-multiplatform