feat: add support for flashing emulation configs on AGX devkits

Author: therishidesai

flake.nix

@@ -43,6 +43,10 @@
                 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

@@ -80,10 +80,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 = ''
           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";
@@ -61,6 +71,15 @@ in
 
     hardware.nvidia-jetpack.flashScriptOverrides =
       let
+        # 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";
+        };
         # Remove unnecessary partitions to make it more like
         # flash_t194_uefi_sdmmc_min.xml, except also keep the A/B slots on each
         # partition
@@ -96,7 +115,7 @@ in
       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";