I'm starting to look at getting Orin SBSA 1 support up. I've hacked up the linux-sbsa/linux-aarch64 getRedistSystem logic so that Orin goes through linux-sbsa and it works! But, I think that the logic for deciding linux-sbsa vs linux-aarch64 in getRedistSystem might not be enough? I think we'd rather know the CUDA driver version (i.e. 540.5.0 vs 595.78) + capabilities to know which system to use? Using cudaMajorMinorVersion could be sad: if someone is on SBSA Orin, they might want to use SBSA CUDA 12? I don't personally care for this usecase, so if y'all think it's not something to worry about

I've uploaded a PR since code is probably easier to read: https://github.com/NixOS/nixpkgs/pull/528562

I'm starting to look at getting Orin SBSA 1 support up. I've hacked up the linux-sbsa/linux-aarch64 getRedistSystem logic so that Orin goes through linux-sbsa and it works! But, I think that the logic for deciding linux-sbsa vs linux-aarch64 in getRedistSystem might not be enough? I think we'd rather know the CUDA driver version (i.e. 540.5.0 vs 595.78) + capabilities to know which system to use? Using cudaMajorMinorVersion could be sad: if someone is on SBSA Orin, they might want to use SBSA CUDA 12? I don't personally care for this usecase, so if y'all think it's not something to worry about, then we don't have to worry about it

I've uploaded a PR since code is probably easier to read: https://github.com/NixOS/nixpkgs/pull/528562

I'm starting to look at getting Orin SBSA support up. I've hacked up the linux-sbsa/linux-aarch64 getRedistSystem logic so that Orin goes through linux-sbsa and it works! But, I think that the logic for deciding linux-sbsa vs linux-aarch64 in getRedistSystem might not be enough? I think we'd rather know the CUDA driver version (i.e. 540.5.0 vs 595.78) + capabilities to know which system to use? Using cudaMajorMinorVersion could be sad: if someone is on SBSA Orin, they might want to use SBSA CUDA 12? I don't personally care for this usecase, so if y'all think it's not something to worry about, then we don't have to worry about it

I've uploaded a PR since code is probably easier to read: https://github.com/NixOS/nixpkgs/pull/528562

https://github.com/NixOS/nixpkgs/pull/528773

cc Kevin Mittman (UTC-7) Bryan Honof connor (burnt/out) (UTC-8)

{
  description = "DAWG Flake";

  inputs = {
    pyproject-nix.url = "github:pyproject-nix/pyproject.nix";
    pyproject-nix.inputs.nixpkgs.follows = "nixpkgs";
  };
  outputs = {
    nixpkgs,
    pyproject-nix,
    ...
  }: let
    system = "x86_64-linux";
    project = pyproject-nix.lib.project.loadPyproject {
      projectRoot = ./.;
    };
    pkgs = import nixpkgs {
      inherit system;
      config = {
        allowUnfree = true;
        cudaSupport = true;
      };
    };

    python = pkgs.python3;
    cudaPackages = pkgs.cudaPackages;
  in {
    devShells.x86_64-linux.default = let
      arg = project.renderers.withPackages {inherit python;};
      pythonEnv = python.withPackages arg;
    in
      pkgs.mkShell {
        packages =
          [pythonEnv]
          ++ (with pkgs; [
            basedpyright
            black

            ffmpeg
            playerctl
            portaudio
            ripgrep

            gcc
            stdenv.cc.cc.lib

            cudaPackages.cudatoolkit
          ]);

        shellHook = ''
          export CUDA_HOME=${cudaPackages.cudatoolkit}
          export CUDA_PATH=${cudaPackages.cudatoolkit}
          export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/run/opengl-driver/lib:/run/opengl-driver-32/lib
        '';
      };

    packages.x86_64-linux.default = let
      attrs = project.renderers.buildPythonPackage {inherit python;};
    in
      python.pkgs.buildPythonPackage attrs;
  };
}