feat: add support for flashing emulation configs on AGX devkits

Author: therishidesai

default.nix

@@ -123,6 +123,10 @@ let
       value = c;
     }) [
     { som = "orin-agx"; carrierBoard = "devkit"; }
+    { som = "orin-agx"; carrierBoard = "devkit-as-nano-4gb"; }
+    { som = "orin-agx"; carrierBoard = "devkit-as-nano-8gb"; }
+    { som = "orin-agx"; carrierBoard = "devkit-as-nx-8gb"; }
+    { som = "orin-agx"; carrierBoard = "devkit-as-nx-16gb"; }
     { som = "orin-agx-industrial"; carrierBoard = "devkit"; }
     { som = "orin-nx"; carrierBoard = "devkit"; }
     { som = "orin-nano"; carrierBoard = "devkit"; }

modules/default.nix

@@ -63,7 +63,7 @@ in
       };
 
       carrierBoard = mkOption {
-        type = types.enum [ "generic" "devkit" ];
+        type = types.enum [ "generic" "devkit" "devkit-as-nx-8gb" "devkit-as-nx-16gb" "devkit-as-nano-8gb" "devkit-as-nano-4gb" ];
         default = "generic";
         description = lib.mdDoc ''
           Jetson carrier board to target. Can be set to "generic" to target a generic jetson carrier board, but some things may not work.

modules/devices.nix

@@ -9,6 +9,16 @@ let
 
   cfg = config.hardware.nvidia-jetpack;
 
+  # The AGX supports Emulating weaker Jetson SoMs on the devkit:
+  # https://developer.ridgerun.com/wiki/index.php/NVIDIA_Jetson_Orin/Flashing_commands_for_emulation#Flashing_the_board
+  agxDevkitTargetBoards = {
+    "devkit" = "jetson-agx-orin-devkit";
+    "devkit-as-nx-8gb" = "jetson-agx-orin-devkit-as-nx-8gb";
+    "devkit-as-nx-16gb" = "jetson-agx-orin-devkit-as-nx-16gb";
+    "devkit-as-nano-8gb" = "jetson-agx-orin-devkit-as-nano8gb";
+    "devkit-as-nano-4gb" = "jetson-agx-orin-devkit-as-nano4gb";
+  };
+
   nvpModelConf = {
     orin-agx = "${pkgs.nvidia-jetpack.l4t-nvpmodel}/etc/nvpmodel/nvpmodel_p3701_0000.conf";
     orin-agx-industrial = "${pkgs.nvidia-jetpack.l4t-nvpmodel}/etc/nvpmodel/nvpmodel_p3701_0008.conf";
@@ -93,7 +103,7 @@ lib.mkMerge [{
     in
     mkMerge [
       (mkIf (cfg.som == "orin-agx") {
-        targetBoard = mkDefault "jetson-agx-orin-devkit";
+        targetBoard = mkDefault (agxDevkitTargetBoards."${cfg.carrierBoard}");
         # We don't flash the sdmmc with kernel/initrd/etc at all. Just let it be a
         # regular NixOS machine instead of having some weird partition structure.
         partitionTemplate = mkDefault "${pkgs.nvidia-jetpack.bspSrc}/bootloader/t186ref/cfg/flash_t234_qspi.xml";